洛谷1525 关押罪犯

原题地址

https://www.luogu.org/problem/show?pid=1525

加权并查集

题目描述

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

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

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

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

输入输出格式

输入格式：

输入文件的每行中两个数之间用一个空格隔开。第一行为两个正整数N 和M，分别表示罪犯的数目以及存在仇恨的罪犯对数。接下来的M 行每行为三个正整数aj，bj，cj，表示aj 号和bj 号罪犯之间存在仇恨，其怨气值为cj。数据保证1<aj=<=bj<=N ，0 < cj≤1,000,000,000，且每对罪犯组合只出现一次。

输出格式：

共1 行，为Z 市长看到的那个冲突事件的影响力。如果本年内监狱中未发生任何冲突事件，请输出0。

输入输出样例

输入样例#1：

4 6

1 4 2534

2 3 3512

1 2 28351

1 3 6618

2 4 1805

3 4 12884

输出样例#1：

3512

说明

【输入输出样例说明】罪犯之间的怨气值如下面左图所示，右图所示为罪犯的分配方法，市长看到的冲突事件影响力是3512（由2 号和3 号罪犯引发）。其他任何分法都不会比这个分法更优。

【数据范围】对于30%的数据有N≤ 15。对于70%的数据有N≤ 2000，M≤ 50000。对于100%的数据有N≤ 20000，M≤ 100000。

讲真的，在写这道题之前真的不知道加权并查集怎么写……尝试着写了写果断WA……然后参考了题解思路又打了一遍，然而依旧WA……最后参考了正解的代码才AC，满满的挫败感……

解题思路

贯彻落实“敌人的敌人就是我的朋友”的原则，相当于建两棵树，相互对立。将边从大到小排序（以下内容参考洛谷题解区__stdcall）

初始化：每个人都在独立的监狱里（可以初始化为自身），对立监狱为0。

1. 如果x和y都在监狱，且两人在同一监狱：不能再把他们调开，这条边即为答案。

2. 如果x和y都在监狱，并且在对立监狱：不管，直接跳过。

3. 如果x和y都在监狱，并且不在同一监狱，若x不在y的对立监狱中，则把x所在监狱合并到y的对立监狱；若y不在x的对立监狱中，则把y所在的监狱合并到x的对立监狱中。

4.如果x、y一个在监狱中，另一个不在，把不在监狱里的那一个合并到另一个的对立监狱里。

参考代码

#include<iostream>
#include<algorithm>
#include<cmath>
#include<cstdio>
#include<cstring>
#include<set>
using namespace std;
int num=0;
int fa[20086],op[20086];
bool vis[20086];
 
struct mc
{
     intx,y,l;
     booloperator<(const mc b) const
     {
         returnl>b.l;
     }
}e[100001];
 
void put(int x,int y,int l)
{
     num++;
     e[num].x=x;
     e[num].y=y;
     e[num].l=l;
}
 
int gt(int x)
{
     if(fa[x]==x)return x;
     fa[x]=gt(fa[x]);
     returnfa[x];
}
 
int main()
{
     intn,m;
     cin>>n>>m;
     intx,y,z,f1,f2;
     for(int i=1;i<=m;i++)
     {
         scanf("%d%d%d",&x,&y,&z);
         put(x,y,z);
     }
     sort(e+1,e+num+1);
     inta=0,b=-1;
     for(int i=1;i<=n;i++)
     {
         fa[i]=i;
         op[i]=0;
     }
     for(int i=1;i<=num;i++)
     {
         x=e[i].x;
         y=e[i].y;
         intf1=gt(x);
         intf2=gt(y);
         if(f1==f2)
         {
              cout<<e[i].l;
              return0;
         }
         if(op[x]==0) op[x]=y;
         else
         {
              f1=gt(y);
              f2=gt(op[x]);
              if(f1!=f2) fa[f1]=f2;
         }
         if(op[y]==0) op[y]=x;
         else
         {
              f1=gt(x);
              f2=gt(op[y]);
              if(f1!=f2) fa[f1]=f2;
         }
     }
     printf("0\n");
     return0;
}