CSL 的魔法 并查集

链接：https://ac.nowcoder.com/acm/contest/551/E
来源：牛客网
“新智认知”杯上海高校程序设计竞赛暨第十七届上海大学程序设计春季联赛

题目描述

有两个长度为 n 的序列，a0,a1,…,an−1和 b0,b1,…,bn−1。CSL 有一种魔法，每执行一次魔法，可以任意挑选一个序列并任意交换序列中两个元素的位置。CSL 使用若干次魔法，得到最终的序列 a 和 b，并且想要让 a0b0+a1b1+…+an−1bn−1的值最小化。求解 CSL 至少使用多少次魔法，能够达到最小化的目标。

输入描述:

第一行有一个整数 n，表示序列的长度。

接下来两行，每行有 n 个整数，分别表示初始序列 a 和 b。

输入数据保证每个序列里的数两两不同。

2≤n≤10^5
1≤ai,bi≤10^9

输出描述:

在一行输出一个整数，表示最少使用的魔法次数。

输入

2
1 2
1 2

输出

1

输入

2
1 2
2 1

输出

0

题解

显然，交换以后a数组最大值对应b数组的最小值，a数组第二大对应b素组第2小... a数组第k大对应b数组第k小。

交换后a数组和b数组不一定有序，只需a数组第k大和b数组第k小一 一对应即可。

我们用map映射一下a数组交换以后每个元素应该在哪个位置，即下标。

如果a数组第i个元素应该交换到j ，那么 将i个j连接起来，假设一个环内有x个元素，那么应该交换x-1次

比如   2 4 6 1 5 3  交换成 1 2 3 4 5 6的顺序

存在 1-2-4    3-6   5  三个环  

1-2-4 这个环有3个元素，需要交换两次；

2-6 这个环有两个元素，需要交换1次；

5这个环只有一个元素，不需要交换

所以用并查集记录连通块个数，求出每个环交换的次数，加起来就可以了

代码

#include<algorithm>
#include <iostream>
#include<cstring>
#include <cstdio>
#include<map>
using namespace std;
typedef long long ll;
const int maxn=1e5+10;
int a[maxn];
int aa[maxn];
int b[maxn];
int father[maxn];
int num[maxn];
map<int,int> m,mm;
int get(int x){
    return x==father[x]?x:father[x]=get(father[x]);
}
int main(){
     int n;
     cin>>n;
     for(int i=0;i<n;i++) father[i]=i,num[i]=1;
     for(int i=0;i<n;i++){
        scanf("%d",a+i);
        aa[i]=a[i];  //将数组a复制一份
     }
     for(int i=0;i<n;i++){
        scanf("%d",b+i);
        mm[b[i]]=i;   //数组中不存在相同的元素，那么用map记录每个数字所在位置
     }
     sort(a,a+n);
     sort(b,b+n);
     for(int i=0;i<n;i++){
        m[a[i]]=mm[b[n-1-i]];   //求出a[i]交换后应该在哪个位置，即下标
     }
     int cnt=0;
     for(int i=0;i<n;i++){
        int x=m[aa[i]];    //第i个数应该交换到x
        if(x==i) continue; 
        //并查集将i，x合并，并更新连通块内的元素个数
        //连通块内的元素个数在根节点处记录,子节点处记录值都为0
        int c=get(i);
        int d=get(x);
        if(c!=d){
            father[d]=c;
            num[c]+=num[d];
            num[d]=0;
        }
     }
     //num[i]不为0，那么i是根节点，num为0不用处理
     for(int i=0;i<n;i++) if(num[i]) cnt+=num[i]-1;
     cout<<cnt<<endl;
    return 0;
}