poj3723

最新推荐文章于 2019-09-29 11:10:00 发布

Stayaccept

最新推荐文章于 2019-09-29 11:10:00 发布

阅读量684

点赞数 1

分类专栏：图论 ---------并查集

本文链接：https://blog.csdn.net/stay_accept/article/details/48660741

版权

图论同时被 2 个专栏收录

72 篇文章 0 订阅

订阅专栏

---------并查集

11 篇文章 0 订阅

订阅专栏

链接：点击打开链接

题意：征召N名女兵，M名男兵，每征召一个人花费10000美金，其中一些男兵女兵关系密切，则征召每个人的费用为10000-（已经征召的人中和自己亲密度的最大值），要求通过适当的征召顺序使得征召所有人的总费用最少，输出总费用

代码：

#include <iostream>
#include <cstdio>
#include <stdlib.h>
#include <cstring>
#include <algorithm>
using namespace std;
int father[50005];
int n,m,r;
struct node{
    int u,v,cost;
}edge[50005];
int cmp(struct node a,struct node b){
    return a.cost<b.cost;
}
int found(int x){
    if(x!=father[x])
    father[x]=found(father[x]);
    return father[x];
}
void unite(int x,int y){
    x=found(x);
    y=found(y);
    if(x==y)
    return;
    father[x]=y;
}
bool same(int x,int y){
    return found(x)==found(y);
}
int kruskal(){
    int i,ans;
    struct node e;
    ans=0;
    sort(edge,edge+r,cmp);
    for(i=0;i<r;i++){
        e=edge[i];
        if(!same(e.u,e.v)){
            unite(e.u,e.v);
            ans+=e.cost;
        }
    }
    return ans;
}                                           //kruskal模板
int main(){                                 //将每个人的费用取反，则变为求最小生成森林
    int t,i;
    scanf("%d",&t);
    while(t--){
        scanf("%d%d%d",&n,&m,&r);
        for(i=0;i<50005;i++)
        father[i]=i;
        for(i=0;i<r;i++){
            scanf("%d%d%d",&edge[i].u,&edge[i].v,&edge[i].cost);
            edge[i].v+=n;                   //并查集，将女兵的编号加N
            edge[i].cost*=-1;
        }
        printf("%d\n",10000*(n+m)+kruskal());
    }
    return 0;
}