wikioi-天梯-提高一等-并查集-1069:关押罪犯

S 城现有两座监狱,一共关押着N 名罪犯,编号分别为1~N。他们之间的关系自然也极

不和谐。很多罪犯之间甚至积怨已久,如果客观条件具备则随时可能爆发冲突。我们用“怨

气值”(一个正整数值)来表示某两名罪犯之间的仇恨程度,怨气值越大,则这两名罪犯之

间的积怨越多。如果两名怨气值为c 的罪犯被关押在同一监狱,他们俩之间会发生摩擦,并

造成影响力为c 的冲突事件。

每年年末,警察局会将本年内监狱中的所有冲突事件按影响力从大到小排成一个列表,

然后上报到S 城Z 市长那里。公务繁忙的Z 市长只会去看列表中的第一个事件的影响力,

如果影响很坏,他就会考虑撤换警察局长。

在详细考察了N 名罪犯间的矛盾关系后,警察局长觉得压力巨大。他准备将罪犯们在

两座监狱内重新分配,以求产生的冲突事件影响力都较小,从而保住自己的乌纱帽。假设只

要处于同一监狱内的某两个罪犯间有仇恨,那么他们一定会在每年的某个时候发生摩擦。那

么,应如何分配罪犯,才能使Z 市长看到的那个冲突事件的影响力最小?这个最小值是少?

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

接下来的M 行每行为三个正整数aj,bj,cj,表示aj 号和bj 号罪犯之间存在仇恨,其怨气值为cj。数据保证且每对罪犯组合只出现一次。

共1 行,为Z 市长看到的那个冲突事件的影响力。如果本年内监狱

中未发生任何冲突事件,请输出0。

4 6

1 4 2534

2 3 3512

1 2 28351

1 3 6618

2 4 1805

3 4 12884

3512

罪犯之间的怨气值如下面左图所示,右图所示为罪犯的分配方法,市长看到的冲突事件

影响力是3512(由2 号和3 号罪犯引发)。其他任何分法都不会比这个分法更优。

【数据范围】

对于30%的数据有N≤ 15。

对于70%的数据有N≤ 2000,M≤ 50000。

对于100%的数据有N≤ 20000,M≤ 100000。

类型:图论  难度:2

题意:有一些罪犯,给出一些两个罪犯之间的怨气值,现在要求把这些罪犯分到两个监狱中,且每个监狱的怨气值取在其中的罪犯的最大怨气值,问怎么分配罪犯,使两个监狱的最大怨气值最小。

分析:并查集题目,一开始想的有点绕,因为给出两个罪犯的怨气值,则两个罪犯最好不在一个监狱中,即给出的是不等价的关系,故解题步骤如下:

1、先将给出的数据按怨气值从大到小排序

2、从大到小遍历数据,用opp[i]代表和罪犯i相斥的罪犯,分情况讨论,设当前数据的罪犯为a,b,怨气值为c:

(1)若a,b均无相斥罪犯,置opp[a]=b,opp[b]=a,将彼此作为相斥罪犯

(2)若a有相斥b没有,那么opp[a]和b就在同一所监狱,合并opp[a]和b为一个等价类,并且,置opp[b]=a

(3)若b有相斥a没有,情况类似(2)

(4)若a,b都有相斥,那么看a,b是否在一个等价类中,如果在,那么当前的c就是最大的怨气值,返回c(因为是按从大到小遍历);若不在一个等价类,那么a和opp[b],b和opp[a]在一个等价类,合并,然后置opp[b]=a,opp[a]=b。

(5)继续遍历下一组数据

代码:

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

int n,m,p;
int fa[20010],path[100010][3],opp[20010];

int mfind(int a)
{
    if(fa[a] != a) fa[a] = mfind(fa[a]);
    return fa[a];
}

void mmerge(int a,int b)
{
    fa[mfind(a)] = mfind(b);
}

int cmp(const void*a, const void*b)
{
    return ((int*)b)[2]-((int*)a)[2];
}

int fun(int x)
{
    if(x>=m) return 0;
    int a = path[x][0];
    int b = path[x][1];
    int c = path[x][2];
    int faa = mfind(a);
    int fab = mfind(b);
    int opa = opp[a];
    int opb = opp[b];

    int ans = 0;
    if(!opa && !opb)
    {
        opp[a] = b;
        opp[b] = a;
    }
    else if(opa && !opb)
    {
        mmerge(opa,b);
        opp[b] = a;
    }
    else if(!opa && opb)
    {
        mmerge(a,opb);
        opp[a] = b;
    }
    else
    {
        if(faa == fab)
        {
            return c;
        }
        else
        {
            mmerge(a,opb);
            mmerge(b,opa);
        }
    }
    return fun(x+1);
}

int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1; i<=n; i++)
        fa[i] = i;
    int a,b;
    for(int i=0; i<m; i++)
    {
        for(int j=0; j<3; j++)
            scanf("%d",&path[i][j]);
    }
    qsort(path,m,sizeof(int)*3,cmp);
    
    memset(opp,0,sizeof(opp));
    printf("%d\n",fun(0));
}


你可能感兴趣的:(并查集,WIKIOI,天梯)