【并查集】牛客比赛NC20210205G 糖果

链接：https://ac.nowcoder.com/acm/contest/9983/G
来源：牛客网

一、题目简介

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

输入描述:
第一行以空格分隔的两个整数n,m。
第二行以空格分隔的n个正整数a_{i}。
接下来m行每行以空格分隔的两个正整数u,v，代表u是v的朋友,v是u的朋友。1≤n≤10^6
0≤m≤10^6,
1≤a{i}≤10^9,u≠v,1≤u,v≤n
输出描述:
购买的最少糖果数以保证每个小朋友都不会不开心。

示例1
输入

3 1
1 2 3
1 2

输出

7

说明
给第三个小朋友买3个糖果，前两个小朋友都买2两个糖果，总共至少买7个糖果。注意如果给第一个小朋友只买了1个糖果，那么他在看到自己的好朋友2有2个糖果的情况下，他就会不开心。

二、题目分析

看题目很容易知道这是并查集，在这里只需要说两点需要注意的事情：
（1）可以在合并时顺便更新值（a[p]=a[q]=max(a[p],a[q]);）
，且可以保证集合中最大值为根节点
（2)合并时只是两两合并，累加结果时必须再次调用find_set()，把集合压缩成一层，都指向根结点。

代码

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
ll  f[1000010],n,m;
ll ans=0,a[1000010];

int find_set(int x){
    return x==f[x]?x:f[x]=find_set(f[x]); 
}

int main()
{
  cin>>n>>m;
  for(int i=1;i<=n;i++){
      scanf("%lld",&a[i]);
      f[i]=i;
  }
  for(int i=1;i<=m;i++){
      int u,v;
      scanf("%d %d",&u,&v);
      int p=find_set(u),q=find_set(v);
      f[p]=q;
      a[p]=a[q]=max(a[p],a[q]);
    }

  for(int i=1;i<=n;i++){
      ans+=a[find_set(i)]; //全局败笔：ans+=a[f[i]]; -_-|||
      //cout<<i<<' '<<f[i]<<' '<<a[i]<<' '<<a[find_set(i)]<<endl;
  }
  cout<<ans<<endl;
  
  return 0;
}