AcWing 257. 关押罪犯 二分图 染色法

Acwing257关押罪犯

题目链接

Meaning

给定两座监狱，其中有 $N$ 名罪犯，编号为 $1\sim n$ ，两名罪犯之间的怨气值为 $c$ 。题目通过给出关系对 $a,b,c$ 来表示罪犯 $a,b$ 之间会产生怨气值为 $c$ 的矛盾，我们要将罪犯分到两个监狱，使得同一个监狱中的恩怨值的最大值最小

Solution

• 将 $a,b,c$ 看成点，点，边权。罪犯之间的恩怨相当于在他们之间连一条边
• 考虑一条数轴，假定一个答案 $x$，在恩怨值大于答案的关系分到两个监狱，否则可以在一个监狱
• 这样，大于这个答案 $x$ 的所有边都是二分图，小于 $x$ 的任何数都不满足这样的性质
• 通过二分 $x$ 加上判断以 $x$ 为界的图是不是二分图得到答案

Code

#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
typedef unsigned long long ULL;
typedef pair<int, int> PII;
#define debug(a) cout << #a << " " << a << endl
const int maxn = 1500;
const int N = 2e5 + 7, M = N * 2;
const int inf = 0x3f3f3f3f;
const long long mod = 1e9 + 7;
inline long long read();

int h[N], e[M], ne[M], w[M], idx;
int n, m;
int color[M];

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]) {
		if(w[i] <= mid) continue;
		int j = e[i];
		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] == 0) {
			if(!dfs(i, 1, mid)) return false; //初始染色
		}
	}
	return true;
}



int main() {

//	freopen("input.txt", "r", stdin);
//	freopen("output.txt", "w", stdout);

	ios::sync_with_stdio(false);
	cin >> n >> m;
	memset(h, -1, sizeof(h));
	while(m--) {
		int a, b, c;
		cin >> 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;
	}

	cout << l << '\n';




	return 0;
}


/*
数组开够了吗 开到上界的n+1次方
初始化了吗
*/







inline LL read() {
	char ch = getchar();
	LL p = 1, data = 0;
	while(ch < '0' || ch > '9') {
		if(ch == '-')p = -1;
		ch = getchar();
	}
	while(ch >= '0' && ch <= '9') {
		data = data * 10 + (ch ^ 48);
		ch = getchar();
	}
	return p * data;
}