洛谷 P1525 关押罪犯 【二分图染色法+二分】

题目链接：https://www.luogu.org/problem/P1525

AC代码：

/*
二分答案，再用二分图染色法判断是否有矛盾，即两个当前点的颜色和与他连接的点的颜色相同；
*/
#include<bits/stdc++.h>
#define max(a,b) a>b?a:b
#define min(a,b) a>b?b:a
using namespace std;
typedef long long ll;
const int inf=0x3f3f3f3f;
const int mod=998244353;
const int N=1e5+10;

struct node
{
    int v,ne,w;
}q[N*2];
int f[N],color[N],e,m,n;

void add(int a,int b,int c)
{
    q[e].v=b;
    q[e].w=c;
    q[e].ne=f[a];
    f[a]=e++;
}

bool judge(int x)
{
    queue<int>que;
    for(int i=1;i<=n;i++) color[i]=0;
    for(int i=1;i<=n;i++)
    {
        if(color[i]==0)
        {
            que.push(i);
            color[i]=1;
            while(!que.empty())
            {
                int u=que.front();
                que.pop();
                for(int j=f[u];j!=-1;j=q[j].ne)
                {
                    if(q[j].w>x)
                    {
                        int v=q[j].v;
                        if(color[v]==0)///如果与当前点连接的点没有颜色，将它染色
                        {
                            if(color[u]==1) color[v]=2;
                            else color[v]=1;
                            que.push(v);
                        }
                        else if(color[v]==color[u])///如果有颜色并且颜色相同，就相当于有矛盾
                            return false;
                    }
                }
            }
        }
    }
    return true;
}

int main()
{
    scanf("%d %d",&n,&m);
    int a,b,c,l=0,r=0,ans;
    for(int i=0;i<=n;i++) f[i]=-1;
    e=0;
    for(int i=0;i<m;i++)
    {
        scanf("%d %d %d",&a,&b,&c);
        add(a,b,c);
        add(b,a,c);
        r=max(r,c);///最大值就是右端点
    }
    while(l<=r)
    {
        int mid=(l+r)/2;
        if(judge(mid))
        {
            r=mid-1;
            ans=mid;
        }
        else l=mid+1;
    }
    printf("%d\n",ans);
    return 0;
}