知识点:并查集(不相交集合)
题目链接: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;
}