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

关押罪犯
题目描述
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

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

①:带权的并查集
设点与其父节点的距离
距离为1:表示该点和父节点不在一个监狱
距离为0:表示该点和父节点在同一个监狱

因此
1:如果两个点在同一个集合中,可以通过计算出两个点之间的距离来判断他们所在监狱的关系。
如果距离为奇数,则表示两者不在同一个监狱。
如果距离为偶数,则表示两者在同一监狱。

2.如果两者a, b不在同一集合,则需要合并两个集合。且合并后要求两者之间的距离是奇数,来表示不在同一个监狱,因此可以先算出a, b到各自祖先fa,fb的距离d[a], d[b],设置fa的祖先为fb(或fb的祖先为fa),且距离为1 - d[a] - d[b]。

#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;
}

②:扩展域并查集
假设1-n为各个犯人,(n+1)- 2*n为各个犯人对立面。即:如果犯人a,b不在同一个监狱,则a + n,b;b + n, a在同一个监狱。如果a,b在同一个监狱,则a + n, b;b + n, a不在同一个监狱

#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;
}

③:染色法判定二分图+二分
对于给定的一个发生矛盾的值c,如果大于c的边能构造出二分图,说明答案小于c。如果不能构造出二分图,说明答案大于c。答案具有二段性,二分即可。

#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;
}
  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值