【ACWing】257. 关押罪犯

题目地址：

https://www.acwing.com/problem/content/259/

$S$城现有两座监狱，一共关押着$N$名罪犯，编号分别为$1\sim N$。他们之间的关系自然也极不和谐。很多罪犯之间甚至积怨已久，如果客观条件具备则随时可能爆发冲突。我们用“怨气值”（一个正整数值）来表示某两名罪犯之间的仇恨程度，怨气值越大，则这两名罪犯之间的积怨越多。如果两名怨气值为$c$的罪犯被关押在同一监狱，他们俩之间会发生摩擦，并造成影响力为$c$的冲突事件。每年年末，警察局会将本年内监狱中的所有冲突事件按影响力从大到小排成一个列表，然后上报到$S$城$Z$市长那里。公务繁忙的$Z$市长只会去看列表中的第一个事件的影响力，如果影响很坏，他就会考虑撤换警察局长。在详细考察了$N$名罪犯间的矛盾关系后，警察局长觉得压力巨大。他准备将罪犯们在两座监狱内重新分配，以求产生的冲突事件影响力都较小，从而保住自己的乌纱帽。假设只要处于同一监狱内的某两个罪犯间有仇恨，那么他们一定会在每年的某个时候发生摩擦。那么，应如何分配罪犯，才能使$Z$市长看到的那个冲突事件的影响力最小？这个最小值是多少？

输入格式：
第一行为两个正整数$N$和$M$，分别表示罪犯的数目以及存在仇恨的罪犯对数。接下来的$M$行每行为三个正整数$a_j$，$b_j$，$c_j$，表示$a_j$号和$b_j$号罪犯之间存在仇恨，其怨气值为$c_j$。数据保证$1\le a_j<b_j<N,0<c_j\le 10^9$且每对罪犯组合只出现一次。

输出格式：
输出共$1$行，为$Z$市长看到的那个冲突事件的影响力。如果本年内监狱中未发生任何冲突事件，请输出$0$。

数据范围：
$N\le 20000,M\le 100000$

思路是二分。将每个罪犯视为图中的顶点，罪犯之间的仇恨值视为边。对于某个仇恨值$x$，如果能将罪犯分为两组，使得每组内部的最大边权是小于等于$x$的，那么显然最优解就是小于等于$x$的，否则就是大于$x$的。那么如何判断是否可以分为符合条件的两组呢，这其实等价于在只考虑边权大于$x$的边的情况下，该图的子图是否是二分图（这里的子图的意思是顶点取原顶点，边取权大于$x$的边），可以用染色法来做。我们只需找到符合条件的最小的$x$即可。二分的初始范围是$[0,M]$，$M$是最大仇恨值。注意$0$也要包含进去，如果最后结果是$0$的话，说明图本身就是二分图。代码如下：

#include <iostream>
#include <cstring>
using namespace std;

// 注意要开两倍的边数
const int N = 20010, M = 200010;
int n, m;
int h[N], e[M], ne[M], w[M], idx;
int col[N];

void add(int a, int b, int c) {
    e[idx] = b, ne[idx] = h[a], w[idx] = c, h[a] = idx++;
}

// 从点x开始深搜，将x染色为c，判断一下是否存在合法的染色
bool dfs(int x, int c, int mi) {
	// 将x点染色
    col[x] = c;
    for (int i = h[x]; ~i; i = ne[i]) {
        int j = e[i];
        // 略过小于mi的边
        if (w[i] <= mi) continue;
        // 如果遇到一个之前染过色的点与x同色，说明不存在合法染色，图不是二分图，返回false
        else if (col[j] == c) return false;
        // 颜色合法，略过
        else if (col[j] == (c ^ 1)) continue;
        // 未染色，则DFS进去染色
        else if (!dfs(j, c ^ 1, mi)) return false;
    }

	// 未发生矛盾，返回true
    return true;
}

bool check(int mi) {
    memset(col, -1, sizeof col);
    for (int i = 1; i <= n; i++)
        if (col[i] == -1) 
            if (!dfs(i, 0, mi)) return false;

    return true;
}

int main() {
    scanf("%d%d", &n, &m);

    memset(h, -1, sizeof h);

    int r = 0;
    while (m--) {
        int a, b, c;
        scanf("%d%d%d", &a, &b, &c);
        add(a, b, c), add(b, a, c);
        r = max(r, c);
    }

    int l = 0;
    while (l < r) {
        int mi = l + (r - l >> 1);
        if (check(mi)) r = mi;
        else l = mi + 1;
    }
    
    // 注意到使得大于x的边组成的子图是二分图的x是一定存在的，
    // 其可以取最大权（对应的子图是空图），所以这里出了循环不需要判断
    printf("%d\n", l);

    return 0;
}

时间复杂度$O((m+n)\log r)$，$r$是最大仇恨值，空间$O(n)$。