poj 3723 Conscription

题意：

招募n个男兵和m个女兵，招募一个人的钱是10000，但是男兵与女兵之间存在亲密关系。

如果招募的一个兵与已经招募的兵之间存在亲密关系，那么招募的钱就是10000 -（亲密系数）。

求最少需要多少钱。

思路：

求最大生成森林。

坑：

运算符重载到底怎么回事啊Orz

代码：

#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <iostream>
#include <set>
using namespace std;

struct node
{
    int x,y;
    int w;

    bool operator < (const node& rhs) const
    {
        return w > rhs.w;
    }
} a[50005];

int par[20005];
int ran[20005];

set<int> s;

void init(int n)
{
    for (int i = 0;i <= n;i++)
    {
        par[i] = i;
        ran[i] = 0;
    }

    s.clear();
}

int fin(int x)
{
    if (x == par[x]) return x;
    else return par[x] = fin(par[x]);
}

void unit(int x,int y)
{
    x = fin(x);
    y = fin(y);

    if (x == y) return;

    if (ran[x] < ran[y])
    {
        par[x] = y;
    }
    else
    {
        par[y] = x;

        if (ran[x] == ran[y]) ran[x]++;
    }
}

int main()
{
    int t;

    scanf("%d",&t);

    while (t--)
    {
        int n,m,r;

        scanf("%d%d%d",&n,&m,&r);

        init(m+n);

        for (int i = 0;i < r;i++)
        {
            int x,y,w;

            scanf("%d%d%d",&x,&y,&w);

            x += m;

            a[i].x = x;
            a[i].y = y;
            a[i].w = w;
        }

        sort(a,a+r);

        int ans = 0;

        for (int i = 0;i < r;i++)
        {
            int x = a[i].x,y = a[i].y;

            if (fin(x) == fin(y)) continue;

            ans += 10000 - a[i].w;

            unit(x,y);
        }

        for (int i = 0;i < m + n;i++)
        {
            int x = fin(i);
            s.insert(x);
        }

        ans += s.size() * 10000;

        printf("%d
",ans);
    }

    return 0;
}