关押罪犯 洛谷1525 并查集

题目大意

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

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

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

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

分析

处理的时候，我们先排序，从冲突值最大的先处理。
判断的时候：
如果两人在同一个并查集，就看它们各自到祖先节点的距离，如果相等，则说明它们事先已经被安排在同一个监狱，那么冲突不可避免，直接输出。
如果距离不等，则说明不会有冲突。
如果不在同一个并查集，则合并。

code

#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<string>
#include<algorithm>
#include<stack>
#include<queue>
#include<vector>

using namespace std;

int f[200000];//并查集数组。 
int dis[200000];//到根的距离。 
int n,m;
struct arr{
    int x,y,w;
}edge[1000000];

bool cmp(arr a,arr b)//排序。 
{
  return a.w>=b.w; 
}

void find(int x)//并查集查找。 
{
    if (f[x]==x) return;
    else{
        find(f[x]);
        dis[x]=(dis[x]%2)+(dis[f[x]]%2);
        f[x]=f[f[x]];
    }
}

void merge(int x,int y)//并查集合并。
{
    int x1,y1;
    find(x);
    find(y);
    x1=f[x]; y1=f[y];
    if (dis[x]%2==dis[y]%2)
        dis[x1]=1;
    f[x1]=y1;
}

int edge_m;

void add(int x,int y,int w)
{
    edge_m++;
    edge[edge_m]=(arr){x,y,w};
}

int main()
{
    scanf("%d%d",&n,&m);
    for (int i=1;i<=m;i++)
    {
        int x,y,w;
        scanf("%d%d%d",&x,&y,&w); 
        add(x,y,w);
    }
    for (int i=1;i<=n;i++) f[i]=i;
    sort(edge+1,edge+m+1,cmp);
    int i=0;
    while (i<m)
    {
        i++;
        find(edge[i].x),find(edge[i].y); 
        if (f[edge[i].x]==f[edge[i].y])
            if (dis[edge[i].x]%2==dis[edge[i].y]%2)
            {
                printf("%d",edge[i].w);
                return 0;
            }
        else continue;
        if (f[edge[i].x]!=f[edge[i].y]) merge(edge[i].x,edge[i].y);
    }
    printf("0");
}