P1525 [NOIP2010 提高组] 关押罪犯 题解

P1525 [NOIP2010 提高组] 关押罪犯

思路

采用贪心的思想，应当尽可能地把怨气值大的一对罪犯拆开，假定一个罪犯的敌人是与他在一起怨气值最大的那个罪犯，则一个罪犯一定和它敌人的敌人在同一个监狱。

从怨气值最大的一对罪犯开始逐步往后处理，如果当前罪犯还没有敌人，说明有关该罪犯的信息目前还没有出现，将其敌人设置好即可(因为当前情况下该罪犯不会和任何人产生矛盾，因此此时该罪犯可以在任意监狱中)；如果当前罪犯已有敌人，那么它的位置可以被确定——和它敌人的敌人在同一个监狱。

如果发现该对罪犯已经被分配在同一个监狱中，说明没有更优解，输出怨气值并结束即可。

使用并查集维护每个罪犯的状况。

#include<bits/stdc++.h>
using namespace std;
struct H{
    int a,b,c;
}h[100010];
bool h_cmp(H x,H y){
    return x.c>y.c;
}
int e[20010],pre[20010];
int find(int x){
    if(x==pre[x]) return x;
    pre[x]=find(pre[x]);
    return pre[x];
}
void join(int x,int y){
    pre[find(x)]=find(y);
}
void initialize(){
    for(int i=0;i<20010;i++) pre[i]=i;
}
int main(){
    initialize();
    int N,M;
    scanf("%d%d",&N,&M);
    for(int i=0;i<M;i++){
        scanf("%d%d%d",&h[i].a,&h[i].b,&h[i].c);
    }
    sort(h,h+M,h_cmp);
    for(int i=0;i<M;i++){
        if(find(h[i].a)==find(h[i].b)){
            printf("%d\n",h[i].c);
            return 0;
        }
        if(e[h[i].a]==0) e[h[i].a]=h[i].b;//如果没有设置敌人，那么设置敌人
        else{
            join(h[i].b,e[h[i].a]);
        }//如果已经有敌人，那么b一定和a的敌人的敌人在同一组
        if(e[h[i].b]==0) e[h[i].b]=h[i].a;
        else{
            join(h[i].a,e[h[i].b]);
        }
    }
    printf("0\n");
    return 0;
}