【NOIP】关押罪犯

带权并查集，其实这种并查集的核心就是“向量”

#include<cstdio>
#include<iostream>
#include<algorithm>
using namespace std;
int n,m,p[20001],r[20001];      //0表示在同一监狱，1表示在不同监狱 
struct node{
    int a,b,c;
}d[100000];
int cmp(node x,node y){
    return x.c>y.c;
}
int findp(int x){
    if(p[x]!=x){
        int px=findp(p[x]);     //找到真正的根结点 
        r[x]=(r[p[x]]+r[x])%2;  //更新当前结点x和真正根结点px的关系r[x]，括号里的r[x]是和旧的父结点的关系 
        p[x]=px;
    }
    return p[x];
}
int main()
{
    int ok=0;
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++) p[i]=i;
    for(int i=0;i<m;i++)
        scanf("%d%d%d",&d[i].a,&d[i].b,&d[i].c);
    sort(d,d+m,cmp);
    for(int i=0;i<m;i++){
        int x=d[i].a,y=d[i].b,z=d[i].c;
        int px=findp(x),py=findp(y);
        if(px!=py){                   //两者不在同一集合，需要把x,y放在不同监狱，也就是向量(x,y)为1 
            p[px]=py;                 //据此可以算出(px,py)的值，也就是r[px] 
            r[px]=(1+r[y]-r[x])%2;
        } 
        else{
            if(!((r[y]-r[x]+2)%2)){   //根据向量关系，两者在同一个监狱 
                ok=1;
                printf("%d",z);
                break;
            }
        }
    }
    if(!ok) printf("0");
    return 0;
}