2021牛客寒假算法基础集训营3【G题糖果】（并查集）

知识点：并查集(不相交集合)
题目链接:https://ac.nowcoder.com/acm/contest/9983/G
互为朋友拿到的糖果应该一样多，并且为集里朋友能得到糖果的最大值

Code

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

int n,s[1000005],a[1000005]
//s[]用来表示朋友之间的关系，a[]用来储存能获得的糖果;

//int find(int x){
//	int r=x;
//	while(s[r]!=r)	r=s[r];
//	return r;
//} 
//一开始用的上面的查找方法，超时了。因为每次查找元素都要搜索路径找到
//根结点。可以用路径压缩来优化，如下面在返回的时候的时候把集里所有的
//元素都指向根结点，下次再搜，就能在O(1)时间内得到结果。
int find(int x){  //查找的优化，路径压缩 
	if(x==s[x])return x;
	return s[x]=find(s[x]); 
}

int main(){
	int m,x,y;
	long long ans=0;
	cin>>n>>m;
	for(int i=1;i<=n;i++){
		cin>>a[i];
		s[i]=i;
	}
	while(m--){
		cin>>x>>y;
		int c=find(x),d=find(y);
		s[c]=d;//将互为朋友关系的，合并到一个集里
		a[d]=max(a[c],a[d]);
//维护，将集中的根结点能得到的糖果更新为集里朋友能得到糖果的最大值
	}
	for(int i=1;i<=n;i++)ans+=a[find(i)];
	cout<<ans;
	return 0;
}