关押罪犯（二分+染色）

思路：求最大值最小，我们看能不能二分，我们二分出答案ans，然后我们吧大于等于ans的2个人关在不同的监狱，我们通过染色判断这种分配是否可以，如果在答案为ans可以的话，那么就说明可以继续往小划分，如果不选那就只能增大ans。

代码：

#pragma GCC optimize(2)
#include<bits/stdc++.h>
 
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
#define SIS std::ios::sync_with_stdio(false)
#define space putchar(' ')
#define enter putchar('\n')
#define lson root<<1
#define rson root<<1|1
typedef pair<int,int> PII;
typedef pair<int,PII> PIII;
const int mod=100003;
const int N=2e5+5;
const int inf=0x7f7f7f7f;

int gcd(int a,int b)
{
    return b==0?a:gcd(b,a%b);
}
 
ll lcm(ll a,ll b)
{
    return a*(b/gcd(a,b));
}
 
template <class T>
void read(T &x)
{
    char c;
    bool op = 0;
    while(c = getchar(), c < '0' || c > '9')
        if(c == '-')
            op = 1;
    x = c - '0';
    while(c = getchar(), c >= '0' && c <= '9')
        x = x * 10 + c - '0';
    if(op)
        x = -x;
}
template <class T>
void write(T x)
{
    if(x < 0)
        x = -x, putchar('-');
    if(x >= 10)
         write(x / 10);
    putchar('0' + x % 10);
}
ll qsm(int a,int b,int p)
{
    ll res=1%p;
    while(b)
    {
        if(b&1)
            res=res*a%p;
        a=1ll*a*a%p;
        b>>=1;
    }
    return res;
}


int head[N],nex[N*2],to[N*2],w[N*2];
int n,m;
int tot;
int color[N];
void add(int u,int v,int val)
{
    to[tot]=v;
    w[tot]=val;
    nex[tot]=head[u];
    head[u]=tot++;

}
bool dfs(int u,int c,int mid)
{
    color[u]=c;
    for(int i=head[u];~i;i=nex[i])
    {
        int v=to[i];
        if(w[i]<=mid)continue;
        if(color[v])
        {
            if(color[v]==c)return false;
        }
        else if(!dfs(v,3-c,mid))return false ;
    }
    return true;
}
bool check(int mid)
{
    memset(color,0,sizeof color);
    for(int i=1;i<=n;i++)
    if(color[i]==0) 
     if(!dfs(i,1,mid))return false;
     return true;
}

int main()
{
    //SIS;
    scanf("%d%d",&n,&m);
    memset(head,-1,sizeof head);
    while(m--)
    {
        int a,b,c;
        scanf("%d%d%d",&a,&b,&c);
        add(a,b,c);add(b,a,c);
    }
    int l=0,r=1e9;
    while(l<r)
    {
        int mid=l+r>>1;
        if(check(mid)) r=mid;
        else l=mid+1;
    }
    printf("%d\n",r);
   return 0;

}

标签：二分,head,return,关押,int,染色,ll,mid,color
来源： https://blog.csdn.net/qq_43619680/article/details/114121990