luogu1525【2010提高】关押罪犯(贪心+并查集)

首先按怒气值从大到小排序,我们贪心的尽量让怒气值最大的两人不在一起。
直到没有办法了,两人只能在一起时,输出答案。
怎么维护呢?类似团伙那道题,让我的敌人和我的敌人在一起。直到两者已经在一个集合中时,输出答案。也可以像食物链那道题,做一个带权并查集,d[i]=0表示i与fa[i]不是敌人,1表示i与fa[i]是敌人。

#include 
#include 
#include 
#define N 20010
#define M 100010
inline int read(){
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
    return x*f;
}
int n,m,fa[N],e[N];
struct edge{
    int x,y,val;
}a[M];
inline bool cmp(edge x,edge y){
    return x.val>y.val;
}
inline int find(int x){
    return x==fa[x]?x:fa[x]=find(fa[x]);
}
int main(){
//  freopen("a.in","r",stdin);
    n=read();m=read();
    for(int i=1;i<=n;++i) fa[i]=i;
    for(int i=1;i<=m;++i) a[i].x=read(),a[i].y=read(),a[i].val=read();
    std::sort(a+1,a+m+1,cmp);
    for(int i=1;i<=m;++i){
        int xx=find(a[i].x),yy=find(a[i].y);
        if(xx==yy){printf("%d\n",a[i].val);return 0;}
        if(!e[a[i].x]) e[a[i].x]=a[i].y;
        if(!e[a[i].y]) e[a[i].y]=a[i].x;
        int zz=find(e[a[i].x]);fa[zz]=yy;
        zz=find(e[a[i].y]);fa[zz]=xx;
    }
    puts("0");
    return 0;
}

你可能感兴趣的:(并查集,其他oj,贪心)