BZOJ1119：置换群的轮换计数
题目要求的同上一题，只不过这里给定了原始的序列，结束的序列的每一个元素的下标，更加一般
1 #include<cstdio> 2 #include<algorithm> 3 #define INF 0x7f7f7f7f 4 using namespace std; 5 const int maxn=1000005; 6 int n,cnt,vmin; 7 bool vis[maxn]; 8 int a[maxn],w[maxn],pos[maxn],to[maxn]; 9 int mn[maxn],size[maxn]; 10 long long ans; 11 long long sum[maxn]; 12 int read() 13 { 14 int x=0,f=1;char ch=getchar(); 15 while(ch<'0'||ch>'9'){if(ch=='-')f-=1; ch=getchar();} 16 while(ch>='0'&&ch<='9'){x=x*10+ch-'0'; ch=getchar();} 17 return x*f; 18 } 19 void solve() 20 { 21 for(int i=1;i<=n;i++) vmin=min(vmin,w[i]); 22 for(int i=1;i<=n;i++) 23 if(!vis[i]) 24 { 25 cnt++;mn[cnt]=w[a[i]]; 26 int k=i; 27 while(1) 28 { 29 vis[k]=1; 30 mn[cnt]=min(mn[cnt],w[a[k]]); 31 sum[cnt]+=w[a[k]]; 32 size[cnt]++; 33 k=to[k]; 34 if(k==i) break; 35 } 36 } 37 for(int i=1;i<=cnt;i++) 38 { 39 long long t1=(long long)(size[i]-2)*mn[i]+sum[i]; 40 long long t2=(long long)(size[i]+1)*vmin+sum[i]+mn[i]; 41 ans+=min(t1,t2); 42 } 43 } 44 void init() 45 { 46 vmin=INF; 47 48 } 49 int main() 50 { 51 init(); 52 n=read(); 53 for(int i=1;i<=n;i++) w[i]=read(); 54 for(int i=1;i<=n;i++) a[i]=read(); 55 for(int i=1;i<=n;i++) {int x=read();pos[x]=i;} 56 for(int i=1;i<=n;i++) to[i]=pos[a[i]]; 57 solve(); 58 printf("%lld",ans); 59 return 0; 60 }