Hdu1102 - Constructing Roads - 最小生成树

#include<stdio.h>
#include<algorithm>
using namespace std;
int p[1005];
struct R
{
	int st,ed,l;
}road[1000050];
int cmp(R a,R b)
{
	return a.l==b.l?a.st<b.st:a.l<b.l;
}
int find(int n)
{
	if(n!=p[n])
	{
		p[n]=find(p[n]);
	}
	return p[n];
}
int main()
{
	int n,i,j,k,q,a,b;
	while(scanf("%d",&n)!=EOF)
	{
		k=1;
		for(i=1;i<=1005;i++)
		{
			p[i]=i;
		}
		for(i=1;i<=n;i++)
		{
			for(j=1;j<=n;j++)
			{
				scanf("%d",&road[k].l);
				road[k].st=i;
				road[k++].ed=j;
			}
		}
		scanf("%d",&q);
		for(i=1;i<=q;i++)
		{
			scanf("%d%d",&a,&b);
			a=find(p[a]);
			b=find(p[b]);
			p[a]=b;
		}
		sort(road+1,road+k,cmp);
		int sum=0;
		for(i=1;i<k;i++)
		{
			a=find(p[road[i].st]);
			b=find(p[road[i].ed]);
			if(a!=b&&road[i].l!=0)
			{
				p[a]=b;
				sum+=road[i].l;
			}
		}
		printf("%d\n",sum);
	}
}