NOIP2010关押罪犯

题目来源https://www.luogu.org/problem/show?pid=1525

二分答案，再对此图进行二分图染色，只取长度>mid的边进行处理

保证长度>mid的边相邻的两个节点染上不同颜色

color数组表示颜色，0表示未染色

#include <algorithm>
#include <iostream>
#include <cstring>
#include <sstream>
#include <cstdlib>
#include <string>
#include <cstdio>
#include <cctype>
#include <vector>
#include <ctime>
#include <cmath>
#include <stack>
#include <queue>
#include <set>
#include <map>
using namespace std;
const int maxn=2e4+1;
const int inf=1e9;
struct node{int t,w;};
vector<node> a[maxn];
int color[maxn]={0};
bool inq[maxn]={0};
int main()
{
ios::sync_with_stdio(false);
int n,m;cin>>n>>m;
int r=-inf,l=inf;
for(int i=1;i<=m;i++)
{
int x,y,z;
cin>>x>>y>>z;
node e;e.w=z;
e.t=y;a[x].push_back(e);
e.t=x;a[y].push_back(e);
l=min(z,l);
r=max(r,z);
}
if(l==r)l=0;
while(l<r)
{
int mid=(l+r)/2;
memset(color,0,sizeof(color));
memset(inq,0,sizeof(inq));
int ok=1;
for(int i=1;i<=n;i++)
{
if(color[i]==0)
{
color[i]=1;
queue<int> q;
q.push(i);inq[i]=1;
while(!q.empty())
{
int u=q.front();q.pop();
for(int j=0;j<a[u].size();j++)
{
if(color[a[u][j].t]==0&&a[u][j].w>mid)
{
color[a[u][j].t]=3-color[u];
if(!inq[a[u][j].t])
{
q.push(a[u][j].t);
inq[a[u][j].t]=1;
}
}
else if(color[a[u][j].t]==color[u]&&a[u][j].w>mid)
{
ok=0;break;
}
}
if(!ok)break;
}
if(!ok)break;
}
}
if(ok)r=mid;
else l=mid+1;
}
cout<<l;
return 0;
}