POJ 3723 Conscription【最小生成树】

题意:

征用一些男生和女生,每个应都要给10000元,但是如果某个男生和女生之间有关系,则给的钱数为10000减去相应的亲密度,征集一个士兵时一次关系只能使用一次。

分析:

kruskal求最小生成树,注意男生和女生用偏移量处理。

代码:

#include<iostream>
#include<algorithm>
#include<cstdio>
using namespace std;
struct edge{int u, v, w;};
const int maxn = 20050, maxm = 50005;
edge e[maxm];
int n, m ,R;
int pa[maxn], _rank[maxn];
bool cmp(const edge &e1,const edge &e2)
{
    return e1.w<e2.w;
}
int _find(int a)
{
    if(a == pa[a]) return a;
    else return pa[a] = _find(pa[a]);
}
void unite(int a, int b)
{
    int ra = _find(a), rb = _find(b);
    if(_rank[ra] < _rank[b]) pa[ra] = rb;
    else{
        pa[rb] = ra;
        if(_rank[ra] == _rank[rb]) _rank[ra]++;
    }
    return;
}
int same(int a, int b)
{
    return _find(a) == _find(b);
}
int kruskal(int r)
{
    sort(e, e+r, cmp);
    for(int i = 0; i <  n+m; i++){
       pa[i] = i;
    }
    int res = 0;
    for(int i = 0; i < r; i++){
        if(!same(e[i].u,e[i].v)){
            unite(e[i].u, e[i].v);
            res += e[i].w;
        }
    }
    return res;
}
int main (void)
{
    int c;scanf("%d",&c);
    while(c--){
        fill(_rank, _rank + n + m, 1);
        scanf("%d%d%d",&n,&m,&R);
        int x, y, r;
        for(int i = 0; i < R;i++){
            scanf("%d%d%d",&x, &y, &r);
            e[i] = (edge){x, y + n, -r};
      }
      printf("%d\n",10000*(n+m) + kruskal(R));
    }
}
posted @ 2016-02-11 11:08  zhuyujiang  阅读(117)  评论(0编辑  收藏  举报