洛谷P1525 关押罪犯

先按照怨气值从大到小排序，最大的当然不能让他们在一起了，将他们分到两个监狱，而与当前的合并的是他的敌人的敌人，合并过程中发现竟然到了一个集合里，说明此时的状态不能再分了，输出怨气值即可

#include
#include
#include
using namespace std;
struct node{
    int x,y,z;
}a[100005];
int f[20005],e[20005];
int n,m,r1,r2;
bool cmp(node a,node b){
    return a.z>b.z;
}
int find(int x){
    return f[x]==x?x:f[x]=find(f[x]);
}
void add(int x,int y){
    x=find(x);y=find(y);
    if (x!=y)
    f[x]=y;
}
int main(){
    cin>>n>>m;
    for (int i=1;i<=m;i++)
    scanf("%d%d%d",&a[i].x,&a[i].y,&a[i].z);
    sort(a+1,a+m+1,cmp);
    for (int i=1;i<=n;i++) f[i]=i;
    for (int i=1;i<=m;i++)
    {
        r1=find(a[i].x);r2=find(a[i].y);
        if (r1==r2){
            cout<return 0;
        }
    else 
      {
    if (!e[a[i].x]) e[a[i].x]=a[i].y;
      else add(e[a[i].x],a[i].y);
      if (!e[a[i].y]) e[a[i].y]=a[i].x;
      else add(e[a[i].y],a[i].x);

    }
    }
    cout<<0;
    return 0;
}

做法2 将f数组扩大到2倍。

#include
#include
#include
#include
using namespace std;
int n,m,f[40005],r1,r2;
struct node{
    int a,b,c;
}e[100005];
bool comp(node a,node b){
    return a.c>b.c;
}
int find(int x){
    if (f[x]!=x) f[x]=find(f[x]);
    return f[x];
}
int main(){
    cin>>n>>m;
    for (int i=1;i<=m;i++)
    cin>>e[i].a>>e[i].b>>e[i].c;
    for (int i=1;i<=n*2;i++) f[i]=i;
    sort(e+1,e+m+1,comp);
    for (int i=1;i<=m;i++){
        r1=find(e[i].a);r2=find(e[i].b);
        if (r1==r2) 
        {cout<return 0;}
        f[r2]=find(e[i].a+n);
        f[r1]=find(e[i].b+n);
    }
    cout<<0;
    return 0;
}