洛谷 P1525 [NOIP2010 提高组] 关押罪犯

题目描述

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?≤109，且每对罪犯组合只出现一次。

输出格式

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

输入输出样例

4 6
1 4 2534
2 3 3512
1 2 28351
1 3 6618
2 4 1805
3 4 12884

3512

说明/提示

【输入输出样例说明】罪犯之间的怨气值如下面左图所示，右图所示为罪犯的分配方法，市长看到的冲突事件影响力是 35123512（由 22 号和 33 号罪犯引发）。其他任何分法都不会比这个分法更优。

【数据范围】

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

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

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

分析

由大到小排序每条边的权值，即是先贪冲突大的把两个分开。 每次往对方敌对阵营加，若出现闭环，则冲突不可调和，输出答案（贪心过了，必定最小）

代码

#include
using namespace std;

//标记a敌对势力，下一次如果与a有冲突就往a敌对和并，若依然有冲突，说明冲突无法避免，输出答案

struct Edge
{
    int a;
    int b;
    int c;
}; Edge edge[100005];

int p[20005];
int ag[20005];

int n,m;

int getf(int v)
{
    if(p[v]==v) return v;
    else
    {
        p[v]=getf(p[v]);
        return p[v];
    }
}

void mer(int a,int b)
{
    int af=getf(a);
    int bf=getf(b);
    if(af!=bf)
    {
        p[bf]=af;
    }
}

bool cmp(Edge &a,Edge &b)
{
    return a.c>b.c;
}


int main()
{
    cin>>n>>m;
    for(int i=1;i<=n;i++)
    {
        p[i]=i;
    }
    
    for(int i=1;i<=m;i++)
    {
        cin>>edge[i].a>>edge[i].b>>edge[i].c;
    }
    
    sort(edge+1,edge+m+1,cmp);
    
    for(int i=1;i<=m+1;i++)
    {
        if(getf(edge[i].a)==getf(edge[i].b))
        {
            cout<endl;
            break;
        }
        else
        {
            if(!ag[edge[i].a]) ag[edge[i].a]=edge[i].b; //a还没有敌人，即没有第二个集子 
            else mer(ag[edge[i].a],edge[i].b);
            if(!ag[edge[i].b]) ag[edge[i].b]=edge[i].a;//做两遍操作，防止之后读入到反方向 
            else  mer(ag[edge[i].b],edge[i].a);   
        }    
    }
    return 0;
}