POJ3723-Conscription

链接：http://poj.org/problem?id=3723

显然求最小生成树。但该图不一定是连通图，所以每处理一棵树还要找下一棵。为了方便，不妨让女孩的标号为0到n-1，而男孩的标号为n到m-1，这样便可放在统一数组记录。

代码（prim算法）：

#include <iostream>
#include <cstdio>
#include <vector>
#include <queue>
#include <algorithm>
using namespace std;
const int INF=10000;
typedef pair<int,int> P;
int n,m,r,t,f[21000],d[21000];
struct edge
{
    int to,cost;
    edge(int to,int cost):to(to),cost(cost){}
};
vector<edge>G[21000];
void prim(void)
{
    int ans=0;
    for(int i=0;i<n+m;i++)
    {
        if(!f[i])
        {
            priority_queue<P,vector<P>,greater<P> > que;
            que.push(P(10000,i));
            while(!que.empty())
            {
                P p=que.top();que.pop();
                int v=p.second;
                if(d[v]<p.first)continue;
                if(!f[v])ans+=d[v];
                f[v]=1;
                for(int j=0;j<G[v].size();j++)
                {
                    edge e=G[v][j];
                    if(d[e.to]>e.cost&&e.to!=i)
                    {
                        d[e.to]=e.cost;
                        que.push(P(d[e.to],e.to));
                    }
                }
            }
        }
    }
    printf("%d
",ans);
}
int main(void)
{
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d %d %d",&n,&m,&r);
        for(int i=0;i<n+m;i++)
        {
            d[i]=INF;
            f[i]=0;
            G[i].clear();
        }
        for(int i=1;i<=r;i++)
        {
            int u,v,d;
            scanf("%d %d %d",&u,&v,&d);
            G[u].push_back(edge(v+n,10000-d));
            G[v+n].push_back(edge(u,10000-d));
        }
        prim();
    }
}