关押罪犯 扩展域并查集 带权并查集 二分图+二分

关押罪犯
题目描述
S 城现有两座监狱，一共关押着 N 名罪犯，编号分别为1~N。

他们之间的关系自然也极不和谐。

很多罪犯之间甚至积怨已久，如果客观条件具备则随时可能爆发冲突。

我们用“怨气值”（一个正整数值）来表示某两名罪犯之间的仇恨程度，怨气值越大，则这两名罪犯之间的积怨越多。

如果两名怨气值为 c 的罪犯被关押在同一监狱，他们俩之间会发生摩擦，并造成影响力为 c 的冲突事件。

每年年末，警察局会将本年内监狱中的所有冲突事件按影响力从大到小排成一个列表，然后上报到 S 城 Z 市长那里。

公务繁忙的 Z 市长只会去看列表中的第一个事件的影响力，如果影响很坏，他就会考虑撤换警察局长。

在详细考察了 N 名罪犯间的矛盾关系后，警察局长觉得压力巨大。

他准备将罪犯们在两座监狱内重新分配，以求产生的冲突事件影响力都较小，从而保住自己的乌纱帽。

假设只要处于同一监狱内的某两个罪犯间有仇恨，那么他们一定会在每年的某个时候发生摩擦。那么，应如何分配罪犯，才能使 Z 市长看到的那个冲突事件的影响力最小？这个最小值是多少？

输入格式
第一行为两个正整数 N 和 M，分别表示罪犯的数目以及存在仇恨的罪犯对数。

接下来的 M 行每行为三个正整数aj，bj，cj，表示aj号和bj号罪犯之间存在仇恨，其怨气值为cj。

数据保证1≤aj<bj<N,0<cj≤1000000000 且每对罪犯组合只出现一次。

输出格式
输出共1行，为 Z 市长看到的那个冲突事件的影响力。

如果本年内监狱中未发生任何冲突事件，请输出0。

数据范围
N≤20000,M≤100000

输入样例：
4 6
1 4 2534
2 3 3512
1 2 28351
1 3 6618
2 4 1805
3 4 12884

输出样例：
3512

由于最近比较忙，等有空了再写题解。先贴三种方法的代码。

①：带权的并查集

#include <iostream>
#include <algorithm>
#include <cstring>

using namespace std;
int n, m, res;
const int N = 2e4 + 10;
const int M = 1e5 + 10;
int p[N], d[N];

int find(int a)
{
    if (p[a] == a)
        return a;
    else
    {
        int fa = p[a];
        p[a] = find(p[a]);
        d[a] += d[fa];
    }
    return p[a];
}

struct P
{
    int a, b, w;
    bool operator < (const P & W) const
    {
        return w > W.w;
    }
}person[M];

int main()
{
    cin >> n >> m;
    for (int i = 0; i < m; i ++)
    {
        int a, b, w;
        cin >> a >> b >> w;
        person[i] = {a, b, w};
    }
    sort(person, person + m);
    
    for (int i = 0; i <= n; i ++)    p[i] = i;
    
    for (int i = 0; i < m; i ++)
    {
        int a = person[i].a, b = person[i].b;
        int fa = find(person[i].a);
        int fb = find(person[i].b);
        
        if (fa == fb && ((d[a] + d[b]) % 2 == 0))
        {
            res = person[i].w;
            break;
        }
        else if (fa != fb)
        {
            p[fb] = fa;
            d[fb] = 1 - d[a] - d[b];
        }
    }
    cout << res << endl;
}

②：扩展域并查集

#include <iostream>
#include <algorithm>
#include <cstring>

using namespace std;
int n, m, res;
const int N = 4e4 + 10;
const int M = 1e5 + 10;
int p[N];

int find(int a)
{
    if (p[a] == a)
        return a;
    return p[a] = find(p[a]);
}

struct P
{
    int a, b, w;
    bool operator < (const P & W) const
    {
        return w > W.w;
    }
}person[M];

int main()
{
    cin >> n >> m;
    for (int i = 0; i < m; i ++)
    {
        int a, b, w;
        cin >> a >> b >> w;
        person[i] = {a, b, w};
    }
    sort(person, person + m);
    
    for (int i = 0; i <= 2 * n; i ++)    p[i] = i;
    
    for (int i = 0; i < m; i ++)
    {
        
        int fa = find(person[i].a);
        int fb = find(person[i].b);
        if (fa == fb)
        {
            res = person[i].w;
            break;
        }
        else
        {
            int a = person[i].a;
            int b = person[i].b;
            p[fa] = find(b + n);
            p[fb] = find(a + n);
        }
    }
    cout << res << endl;
}

③：染色法判定二分图+二分

#include <iostream>
#include <cstring>

using namespace std;
const int N = 2e4 + 10;
const int M = 2e5 + 10;
int n, m;
int col[N], h[N], e[M], ne[M], w[M], idx;

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

bool dfs(int x, int c, int mid)
{
    col[x] = c;
    for (int i = h[x]; ~i; i = ne[i])
    {
        if (w[i] <= mid) continue;
        
        int j = e[i];
        if (col[j] == -1){   
            if (!dfs(j, 3 - c, mid)) return false; }
        
        else if (col[j] == c) return false;
    }
    return true;
}

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

int main()
{
    memset(h, -1, sizeof h);
    cin >> n >> m;
    for (int i = 0; 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 < r)
    {
        int mid = l + r >> 1;
        if (check(mid)) r = mid;
        else l = mid + 1;
    }
    
    cout << l;
}

标签：二分,const,int,查集,person,带权,罪犯,return,find
来源： https://blog.csdn.net/aaaaa124421421/article/details/111403425