1067 Sort with Swap(0, i) (25 point(s)并查集做法

思路

对于a[i] = j 的时候，将i和j加入到同一个集合中

对于一个集合，如果该集合中有0，则只需要交换集合个数cnt-1，因为0是自动归位的
如果该集合中没有0，就需要拿一个数跟0交换（交换次数+1），然后这个集合就有0了，交换次数为集合个数cnt-1，然后还需要将0从哪拿的再放回去（交换次数+1）

所以并查集维护一下就好了，用一个记录集合元素个数会更加好写

#include <iostream>
#include <stdio.h>
#include <algorithm>
#include <string>
#include <cstring>
#include <queue>
#include <cmath>
#include <stack>
#include <map>
#include <unordered_map>
#include <deque>
using namespace std;
#define debug_in freopen("in.txt","r",stdin);
#define debug_out freopen("out.txt","w",stdout);
typedef long long ll;

void pt(){cout<<'\n';}
void read(int &x){scanf("%d",&x);}
void read(ll &x){scanf("%lld",&x);}
template<class H, class ... T> void pt(H h,T ... t){ cout<<" "<<h; pt(t...);}
template<class H, class ... T> void read(H h,T ...t) {read(h); read(t...);}

int N;
const int maxn = 1e6+10;
int a[maxn];
int fa[maxn];
int find(int x){
    return x == fa[x] ? x : fa[x] = find(fa[x]);
}
int sz[maxn];
int main(){
//    debug_in;
//    debug_out;

    read(N);
    for(int i = 0;i<N;i++) read(a[i]);
    for(int i = 0;i<N;i++) fa[i] = i,sz[i] = 1;
    for(int i = 0;i<N;i++){
        int fx = find(a[i]),fy = find(i);
        if(fx != fy){
            fa[fx] = fy;
            sz[fy] += sz[fx];
        }
    }
    int ans = 0;
    for(int i = 0;i<N;i++) {
        if(find(i) == i && i != a[i]) {
            ans += sz[i] + 1;
        }
    }
    if(a[0] != 0) ans -= 2;
    cout<<ans<<'\n';

    return 0;
}