洛谷 P1525 关押罪犯(并查集)

时间:2022-01-22 14:36:58

传送门


在两个有冲突的罪犯中间连边,边权为怒气值,把边按权排序,然后用并查集处理。
我们让边权大的边得两个点不在同一集合中,一旦遇到冲突,当前边权即是最大怒气值。
Code:

#include<cstdio>
#include<cstdlib>
#include<algorithm>
using namespace std;

struct node{
    int x,y,c;
    inline bool operator<(const node&b)const
    {
        return c>b.c;
    }
}a[100010];
int n,m,f[20010],b[20010];

inline int findfa(int x)
{
    if(f[x]==x) return x;
    else return f[x]=findfa(f[x]);
}

int main()
{
    scanf("%d %d",&n,&m);
    for(int i=1;i<=n;i++) f[i]=i;
    for(int i=1;i<=m;i++)
        scanf("%d %d %d",&a[i].x,&a[i].y,&a[i].c);
    sort(a+1,a+1+m);
    for(int i=1;i<=m+1;i++)
    {
        int fx=findfa(a[i].x),fy=findfa(a[i].y);
        if(fx==fy)
        {
            printf("%d",a[i].c);
            return 0;
        }
        else
        {
            if(!b[a[i].x]) b[a[i].x]=a[i].y;
            else f[findfa(b[a[i].x])]=fy;
            if(!b[a[i].y]) b[a[i].y]=a[i].x;
            else f[findfa(b[a[i].y])]=fx;
        }
    }
    return 0;
}