并查集 牛客练习赛41 C抓捕盗窃犯

题目链接 ：https://ac.nowcoder.com/acm/contest/373/C

题意，初始每一个城市都有一伙盗贼，没过一个时刻盗贼就会逃窜到另一个城市，你可以在m个城市设置监察站，会逮捕所有经过该城市的盗贼

分析：仔细分细题目，因为每个城市的盗贼都是流动的，这就可能会形成环，而如果成环的话，在环所在任一城市都可以把这批城市的全部盗贼逮捕，就不需要再环上设多个监察站了，进一步推广，因为可以自己往自己的城市跑，所以也有可能是链，而链也是满足设一个监察站可以逮捕所有

自然想到引入并查集，用map来保存

#include <bits/stdc++.h>
using  namespace  std;
typedef long long ll;
const int maxn=1e5+7;
const int mod=1e9+7;
ll a[maxn],v[maxn];
ll par[maxn];
ll rnk[maxn];
ll c[maxn];
bool cmp(const ll &a,const ll &b){
    return a>b;
}
void init(){
    for(ll i=0;i<maxn;i++) par[i]=i,rnk[i]=0;    
}
ll find(ll x){
    if(par[x]==x){
        return x;
    }
    else{
        return par[x]=find(par[x]);
    }
}
void unite(ll x,ll y){
    x=find(x);y=find(y);
    if(x==y) return ;
    if(rnk[x]<rnk[y]){
        par[x]=y;
    }else {
        par[y]=x;
        if(rnk[x]==rnk[y]) rnk[x]++;
    }
}
bool same(ll x,ll y){
    return find(x)==find(y);
}
int  main(){  
    ll n,m;scanf("%lld%lld",&n,&m);
    init();
    for(int i=1;i<=n;i++){
        scanf("%lld",&a[i]);
    }
    for(int i=1;i<=n;i++){
        scanf("%lld",&v[i]);
        unite(i,v[i]);
    }
    map<ll,ll> ans;
    map<ll,ll>::iterator ite;
    for(ll i=1;i<=n;i++){
        ans[find(i)]+=a[i];
    }
    //for(int i=1;i<=n;i++) cout<<find(i)<<" ";
    int j=0;
    for(ite=ans.begin();ite!=ans.end();++ite){
        c[j++]=ite->second;
    }
    sort(c,c+ans.size(),cmp);
    ll cnt=0;
    for(ll i=0;i<m&&i<ans.size();i++){
    cnt+=c[i];}//cout<<c[i]<<endl;}
    cout<<cnt<<endl;
    return 0;
}