算法竞赛进阶指南 关押罪犯

题面

题解

我们将罪犯当作点，罪犯之间的仇恨关系当作点与点之间的无向边，边的权重就是罪犯之间的仇恨值，那么原来的问题就转化成了把所有点分成两组，使得各组内的边的权重的最大值最小

我们在[0,1e9] 之间二分枚举最大权值x，当x固定之后，我们就可以判断能否将所有点分成两组，使得权值大于x的边在两组之间，而小于等于x的权值在组内，也就是判断由所有点以及所有权值大于 limitlimit 的边构成的新图是否是二分图。

代码

#include<iostream>
#include<cstdio>
#include<string>
#include<cstring>
#include<algorithm>

using namespace std;
const int N = 2e4 + 10, M = 2e5 + 10;

int n, m;
int h[N], e[M], ne[M], w[M], idx;
int color[N];  //0表示未染色,12表示染两种不同的颜色

//添加a->b的边
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 x) {
    color[u] = c;
    //给所有连通块都染色(树的深度遍历)
    for (int i = h[u]; i != -1; i = ne[i]) {
        if (w[i] <= x) continue;
        int j = e[i];
        if (!color[j]) { //未染色
            if (!dfs(j, 3 - c, x)) { //染色冲突
                return false;
            }
        } else if (color[j] == c) { //染色相同
            return false;
        }
    }
    return true;
}

//判断是否为二分图
bool check(int x) {
    memset(color, 0, sizeof color);
    for (int i = 1; i <= n; i++) {
        if (color[i] == 0) {  //未染色
            if (!dfs(i, 1, x)) { //染色发生冲突
                return false;
            }
        }
    }
    return true;
}


int main() {

    memset(h,-1,sizeof h);
    cin >> n >> m;
    for (int i = 1; i <= m; i++) {
        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 << endl;

    return 0;
}