关押罪犯 洛谷1525 并查集

题目大意

S 城现有两座监狱，一共关押着N 名罪犯，编号分别为1~N。他们之间的关系自然也极不和谐。很多罪犯之间甚至积怨已久，如果客观条件具备则随时可能爆发冲突。我们用“怨气值”（一个正整数值）来表示某两名罪犯之间的仇恨程度，怨气值越大，则这两名罪犯之间的积怨越多。如果两名怨气值为c 的罪犯被关押在同一监狱，他们俩之间会发生摩擦，并造成影响力为c 的冲突事件。

每年年末，警察局会将本年内监狱中的所有冲突事件按影响力从大到小排成一个列表，然后上报到S 城Z 市长那里。公务繁忙的Z 市长只会去看列表中的第一个事件的影响力，如果影响很坏，他就会考虑撤换警察局长。

在详细考察了N 名罪犯间的矛盾关系后，警察局长觉得压力巨大。他准备将罪犯们在两座监狱内重新分配，以求产生的冲突事件影响力都较小，从而保住自己的乌纱帽。假设只要处于同一监狱内的某两个罪犯间有仇恨，那么他们一定会在每年的某个时候发生摩擦。

那么，应如何分配罪犯，才能使Z 市长看到的那个冲突事件的影响力最小？这个最小值是多少？

分析

首先可按照冲突值从大到小排序来进行处理，先解决冲突值最大的纠纷，将两个罪犯放到不同的监狱，将放在同一个监狱的罪犯并到同一颗树上，同时用一个数组存储罪犯的敌人，再次遇到敌人时候可将这两个敌人并到一棵树上，若当前两个罪犯已经在同一颗树上，则该冲突值就是最大冲突值。

code 

#include <bits/stdc++.h>
typedef long long ll;
using namespace std;
const int maxn=1e5+3;
struct node
{
    ll u;
    ll v;
    ll w;
}A[maxn];
ll emp[20003];
ll pre[20003];
ll find(ll x)
{
    if(x!=pre[x])
        pre[x]=find(pre[x]);
    return pre[x];
}
void joint(ll x,ll y)
{
    ll fx=find(x),fy=find(y);
    if(fx!=fy)
        pre[fx]=fy;
}



ll cmp(node x,node y)
{
    return x.w>y.w;
}

int main()
{
    ll n,m;
    cin>>n>>m;
    for(int i=1;i<=m;i++)
    {
        cin>>A[i].u>>A[i].v>>A[i].w;
    }
    sort(A+1,A+m+1,cmp);
    for(int i=1;i<=20005;i++)
        pre[i]=i;
    for(int i=1;i<=m;i++)
    {
       ll p=A[i].u;
       ll q=A[i].v;
        if(find(p)==find(q))
        {
            printf("%lld\n",A[i].w);
            return 0;
        }

        if(!emp[p]&&!emp[q])
        {
            emp[p]=q;
            emp[q]=p;
        }
        else if(emp[p]&&!emp[q])
        {
            joint(emp[p],q);
            emp[q]=p;
        }
        else if(!emp[p]&&emp[q])
        {
            joint(p,emp[q]);
            emp[p]=q;
        }
        else if(emp[p]&&emp[q])
        {
            joint(emp[p],q);
            joint(emp[q],p);
        }
    }
    printf("0\n");
    return 0;
}