NOIP2010提高组 关押罪犯

题目链接

题意就是，把罪犯放到两个不同的监狱中，每个犯人之间可能会有冲突值，如果两个犯人分在同一个监狱内他们的冲突值就会被激活，产生影响力为冲突值的冲突，监狱内的冲突值为所有人之间冲突的最大值，求分配后最小的冲突值

可以把两个监狱看成二分图的两个部分，罪犯之间的冲突值看作是边权，用二分的方法，二分监狱内的冲突值，如果当前二分到的冲突值为mid，也就是说监狱内最多容纳冲突值mid的一对犯人，可以遍历所有的边，将冲突（边权）大于mid的一对犯人分配到两个不同的监狱，即在新图上建立一条两个点之间的边，最后判断新图是否是二分图就可以得出当前的方案是否合法。

判断的时候用染色法判断是否是二分图即可

我一开始染色法写错了QAQ

#include <bits/stdc++.h>
#define mem(a, b) memset(a, b, sizeof a)
using namespace std;
const int N = 200100;
int n, m, nex[N], head[N], edge[N], to[N], cnt;
int vis[N], nex1[N], head1[N], to1[N];
void init() {
	cnt = 0;
	mem(nex, -1);
	mem(head, -1);
	mem(vis, 0);
	mem(nex1, -1);
	mem(head1, -1);
}
void add(int a, int b, int c) {
	cnt++;
	nex[cnt] = head[a];
	head[a] = cnt;
	to[cnt] = b;
	edge[cnt] = c;
}
void add(int a, int b) {
	cnt++;
	nex1[cnt] = head1[a];
	head1[a] = cnt;
	to1[cnt] = b;
}
bool dfs(int p, int color, int fa) {
	bool f = true;
	vis[p] = color;
	for (int i = head1[p]; i != -1; i = nex1[i]) {
		int y = to1[i];
		if (y == fa)continue;
		if (vis[y] == 0)f = f && dfs(y, 3 - color, p);
		else if (vis[y] == vis[p])f = f && false;
		else f = f && true;
	}
	return f;
}
bool check(int mid) {
	bool f = true;
	mem(nex1, -1);
	mem(head1, -1);
	mem(to1, -1);
	mem(vis, 0);
	cnt = 0;
	for (int i = 1; i <= n; i++) {
		for (int j = head[i]; j != -1; j = nex[j]) {
			if (edge[j] > mid) {
				add(i, to[j]);
			}
		}
	}
	for (int i = 1; i <= n; i++) {
		if (vis[i] == 0)f = f && dfs(i, 1, -1);
	}
	return f;
}
int main()
{
	init();
	cin >> n >> m;
	int l = 0, r = 0;
	while (m--) {
		int a, b, c;
		cin >> a >> b >> c;
		add(a, b, c);
		add(b, a, c);
		r = max(r, c);
	}
	int res = 0x3f3f3f3f;
	while (l <= r) {
		int mid = (l + r) / 2;
		if (check(mid)) {
			res = min(res, mid);
			r = mid - 1;
		}
		else {
			l = mid + 1;
		}
	}
	cout << res << "\n";
	return 0;
}