POJ 2421 Constructing Roads（并查集+最小生成树）

//804K 47MS
#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
using namespace std;

const int maxn=110;

int n,ans,tot,m,u,v,w;
int pre[maxn];

struct Edge{
int u,v,w;
}edge[maxn*maxn];

void init()
{
ans=tot=0;
for(int i=0;i<maxn;i++)
pre[i]=i;
}

bool cmp(struct Edge a,struct Edge b)
{
return a.w<b.w;
}

int find(int x)
{
return pre[x]==x?x:pre[x]=find(pre[x]);
}

void mix(int x,int y)
{
int fx=find(x);
int fy=find(y);
if(fx!=fy) pre[fx]=fy;
}

int main()
{
#ifdef LOCAL
freopen("in.txt","r",stdin);
//freopen("out.txt","w",stdout);
#endif
while(~scanf("%d",&n))
{
init();
for(int i=1;i<=n;i++)
{
for(int j=1;j<=n;j++)
{
scanf("%d",&edge[tot].w);
edge[tot].u=i;
edge[tot].v=j;
tot++;
}
}
sort(edge,edge+tot,cmp);
scanf("%d",&m);
for(int i=0;i<m;i++)
{
scanf("%d%d",&u,&v);
mix(u,v);
}
for(int i=0;i<tot;i++)
{
u=edge[i].u;
v=edge[i].v;
w=edge[i].w;
int fu=find(u);
int fv=find(v);
//printf("u=%d fu=%d v=%d fv=%d\n",u,fu,v,fv);
if(fu!=fv)
{
pre[fu]=fv;
ans+=w;
}
}
printf("%d\n",ans);
}
return 0;
}