牛客寒假算法基础训练营3 G.糖果

传送门
在一个幼儿园里面有n个小朋友，分别编号1,2,…,n1,2,…,n。在这些小朋友中有一些小朋友互为朋友关系，总共有\mathit mm对朋友。
作为幼儿园老师，你想买一些糖果分给小朋友，你知道第i个小朋友想要至少想要ai个糖果。
​
个糖果，否则他就会不开心。
同时，如果一个小朋友得到的糖果数小于他某个朋友得到的糖果数，他也会不开心。
请问你最少买多少糖果才能保证每个小朋友都不会不开心呢？

思路：

并查集，一坨好朋友中每个好朋友得到的糖果为这坨好朋友中得糖果最多的人所得到的糖果。

#include<bits/stdc++.h>
using namespace std;
#define ll long long 
const int mod = 1e9 + 7;
ll pre[1010000];
ll a[1000010];
ll find(ll a)
{
	if(pre[a] != a)
	return pre[a] = find(pre[a]);
	return pre[a];
} 
void join(ll x, ll y)
{
	x = find(x),y = find(y);
	pre[x] = y;
	a[y] = max(a[x], a[y]);
} 
int main()
{
	ll n,m;
	scanf("%lld%lld",&n,&m);
	for(ll i = 1; i <= n; i++)
	{
		scanf("%lld",&a[i]);
		pre[i] = i;
	}
	
	while(m--)
	{
		ll v,u;
		scanf("%lld%lld",&v,&u);
		join(v,u);
	}
	ll ans = 0;
	for(int i = 1; i <= n; i++)
	{
		ans += (a[find(i)]);
	}
	printf("%lld\n",ans);
}