AcWing257. 关押罪犯(二分图+染色法)

AcWing257. 关押罪犯(二分图+染色法)_第1张图片

输入样例:

4 6
1 4 2534
2 3 3512
1 2 28351
1 3 6618
2 4 1805
3 4 12884

输出样例:

3512

解析:

        二分,每次查看是否是二分图

#include
using namespace std;
typedef long long ll;
const int N=2e4+5,M=2e5+5;
int n,m;
int h[N],w[M],ne[M],e[M],idx;
int color[N];
void add(int a,int b,int c){
	e[idx]=b,w[idx]=c,ne[idx]=h[a],h[a]=idx++;
}
bool dfs(int u,int c,int mid){
	color[u]=c;
	for(int i=h[u];~i;i=ne[i]){
		int j=e[i];
		if(w[i]<=mid) continue;
		if(color[j]){
			if(color[j]==c) return false;
		}
		else if(!dfs(j,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]){
			 if(!dfs(i,1,mid)) return false;
		}
	}
	return true;
}
int main(){
	scanf("%d%d",&n,&m);
	memset(h,-1,sizeof h);
	for(int i=1;i<=m;i++){
		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>1;
		if(check(mid)) r=mid;
		else l=mid+1;
	}
	cout<

你可能感兴趣的:(AcWing,图论,算法,深度优先,c++,二分图)