重新排列奶牛（并查集，连通块中点的数量

1921. 重新排列奶牛（并查集，连通块中点的数量）
比如 1 2 3 和 3 1 2，经过merge三个元素一定都在一个集合,可以理解为关系为一个环的各个元素，两两合并，最后整个环中的元素都一定在一个集合

#include <bits/stdc++.h>
using namespace std;
#define int long long int

const int N=5e2+5;
int n,s;
int a[N],b[N];
int fa[N];
int num[N];//num[ fa[i] ]:fa[i]该集合中有多少个元素
int find(int x){//寻找a[x]所在的集合的编号
    if(x==fa[x])return x;
    return fa[x]=find(fa[x]);
}
void merge(int x,int y){
    int fx=find(x);
    int fy=find(y);
    if(fx!=fy){
        fa[fx]=fy;
        num[fy]+=num[fx];//记录合并后编号为fy的集合的元素个数
    }
}
signed main(){
    cin>>n;
    for(int i=1;i<=n;i++){
        cin>>a[i],fa[i]=i,num[i]=1;
    }
    for(int i=1;i<=n;i++){
        cin>>b[i];
        if(b[i]!=a[i]){
            merge(a[i],b[i]);
        }
    }
    int res=0;//最后分成了几个集合
    int mx=-1;//含元素最多的集合
    for(int i=1;i<=n;i++){
        if(b[i]!=a[i]&&b[i]==find(b[i])){
// 比如 1 2 3 和  3 1 2，经过merge三个元素一定都在一个集合 
// 我们只要找到 这三个元素合并到的集合的编号，
// 因为需要根据这个编号得到元素数量，
// 而且为了只计数一次集合，只在碰到这个编号时才计数
            res++;
            mx=max(mx,num[b[i]]);
        }
    }
    cout<<res<<" "<<mx;
    return 0;
}

AcWing 4872. 最短路之和（Floyd）

树形dp