CodeForces 893C （并查集板子题）

刷题刷到自闭，写个博客放松一下

题意：n个人，m对朋友，每寻找一个人传播消息需要花费相应的价钱，朋友之间传播消息不需要花钱，问最小的花费

把是朋友的归到一起，求朋友中花钱最少的，将所有最少的加起来。

#include<stdio.h>
#include<algorithm>
#include<string.h>
#define ll long long
using namespace std;

ll a[1000100];

struct lll
{
    ll par,b,c;
}p[100100];

int cmp(lll x,lll y)
{
    if(x.par==y.par) return x.b<y.b;
    return x.par<y.par;
}

ll find(ll x)
{
    if(x!=p[x].par) p[x].par=find(p[x].par);
    return p[x].par;
}

void unionn(ll x,ll y)
{
    ll fa=find(x);
    ll fb=find(y);
    if(fb==fa) return ;
    if(p[fa].c<p[fb].c){
        p[fa].par=fb;
    }
    else p[fb].par=fa;
    if(p[fa].c==p[fb].c) p[fa].c++;
}

int main()
{
    ll n,m;
    while(scanf("%lld%lld",&n,&m)!=EOF){
        memset(a,0,sizeof a);
        memset(p,0,sizeof p);
        for(ll i=1;i<=n;i++){
            scanf("%lld",&a[i]);
            p[i].par=i;
            p[i].b=a[i];
            p[i].c=0;
        }
        ll x,y;
        for(ll i=0;i<m;i++){
            scanf("%lld%lld",&x,&y);
            unionn(x,y);
        }
        ll sum=0;
        for(int i=1;i<=n;i++)
            while(p[i].par!=i&&p[p[i].par].par!=p[i].par)       //是同一类但是头不是老大 把头变为老大；
                p[i].par=p[p[i].par].par;
        sort(p+1,p+n+1,cmp);                //把结构体按类排序，主要是同一类的花费少的在前
        sum=p[1].b;
        for(int i=2;i<=n;i++){
            if(p[i].par!=p[i-1].par) sum+=p[i].b;       //是同一类的加上花费最少的第一个 不是同一类就继续循环
        }
        printf("%lld\n",sum);
    }
    return 0;
}