有环的情况生成并查集

题目链接：牛客练习赛41Chttps://ac.nowcoder.com/acm/contest/373/C

并查集最基本的查找和合并就不多讲了，直接看函数吧。

合并过程中如果出现环，如5->6->7->5，那么在找最终老大时会出现死循环，边输入每个节点的老大信息边合并可以有效的处理这种情况，具体解析看代码注释。

#include<iostream>
#include<cstring>
#include<vector>
#include<algorithm>
using namespace std;
const int N=100005;
int a[N];
int fa[N];//存各自的老大 
long long c[N]; 
vector<long long> s; 
int n,m;
void init(){
	memset(c,0,sizeof(c));	
	for(int i=1;i<=n;i++) fa[i]=i;
	s.clear();
}
int find(int x)
{
	int r = x;
	while (fa[r] != r)//查找根节点(找到老大是谁),如果有环呢 
	{//如5,6,7的环，fa[5]=6,r=6;fa[6]=7,r=7;fa[7]=5,break; 
		/*if(fa[r]==x){
			fa[r]=r;
			break;//有环，老大是最后找的那个
		}这段舍弃不知为啥不行*/			 
		r = fa[r];
	}
	//压缩
	int i = x, j;
	while (fa[i] != r)
	{
		j = fa[i];//保存上一个结点
		fa[i] = r;//压缩当前结点
		i = j;//准备压缩上一个结点
	}
	return r;
}
void mix(int x, int y)//让y成为x的直接老大 
{
	int fx = find(x), fy = find(y); 
	fa[fx] = fy;
}
/*
样例 
8 2
1 2 3 4 1 2 3 12
2 3 3 3 6 7 5 8
*/
void solve(){
	
	//计算每个连通块的流量 
	for(int i=1;i<=n;i++){
		c[find(i)]+=a[i]; 
	}	
	//显示连通块信息 上面find更新后才是最终的 3 3 3 3 7 7 7 8
	/*for(int i=1;i<=n;i++)
		cout<<fa[i]<<" ";
	cout<<endl;*/
	
	for(int i=1;i<=n;i++){
		if(c[i])s.push_back(c[i]);
	}//没有人流量的舍去 
		
	sort(s.begin(),s.end(),greater<long long>());
	
	//输出连通块的流量排名12 10 6
	/*for(int i=0;i<s.size();i++)
		cout<<s[i]<<" ";
	cout<<endl;*/
	
	//加前m个连通块的流量 
	long long ans=0;
	int j=n;
	for(int i=0;i<min(m,(int)s.size());i++){
		ans+=s[i];
	}
	printf("%lld",ans);
} 
int main(){
while(~scanf("%d%d",&n,&m)){
	init();
    for(int i=1;i<=n;i++)
        scanf("%d",&a[i]);
    for(int i=1;i<=n;i++){
    	int x;
    	scanf("%d",&x);
    	mix(i,x);
    	//输入一次,合并一次可以避免环路出现的问题 
        //比如5,6,7这个环。先mix(5,6),find(5)=5,find(6)=6,pre[5]=6;
		//mix(6,7),find(6)=6,find(7)=7,pre[6]=7;
		//mix(7,5),find(7)=7,find(5)=7 
	}        
    solve();
}     
    return 0;
}