当前位置 : 主页 > 编程语言 > c++ >

2010提高组-C]关押罪犯(扩展域并查集

来源:互联网 收集:自由互联 发布时间:2021-06-23
题:https://www.cometoj.com/problem/0073 #includebits/stdc++.h using namespace std; const int M=1e5+ 4 ; struct node{ int u,v,w;}e[M]; int f[M]; bool cmp(node p,node q){ return p.w q.w;} int find( int x){ return f[x]==x?x:f[x]= find(f[x])

题:https://www.cometoj.com/problem/0073

 

#include<bits/stdc++.h>
using namespace std;
const int M=1e5+4;
struct node{
    int u,v,w;
}e[M];
int f[M];
bool cmp(node p,node q){
    return p.w>q.w;
}
int find(int x){
    return f[x]==x?x:f[x]=find(f[x]);
}
int main(){
    int n,m;
    scanf("%d%d",&n,&m);
    for(int i=1;i<=m;i++){
        scanf("%d%d%d",&e[i].u,&e[i].v,&e[i].w);
    }
    sort(e+1,e+1+m,cmp);
    for(int i=1;i<=2*n;i++)
        f[i]=i;
    for(int i=1;i<=m;i++){
        int u=e[i].u,v=e[i].v;
        int uu=e[i].u+n,vv=e[i].v+n;
        int a=find(u),b=find(v);
        int aa=find(uu),bb=find(vv);
        if(a==b){
            return printf("%d\n",e[i].w),0;
        }
        else {
            if(a!=bb)
                f[a]=bb;
            if(aa!=b)
                f[aa]=b;
        }
    }
    printf("0\n");
    return 0;
}
View Code

 

把每个人拆成两个点分别表示与这个人一个监狱的集合和与这个人不同监狱的集合即可。

网友评论