POJ 1258 Agri-Net 【最小生成树】

//9188654	ylwh	1258	Accepted	456K	16MS	G++	1023B	2011-08-18 17:20:32
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <math.h>
#include <algorithm>
using namespace std;
#define N 501
struct edge
{
	int x, y, len;
}e[N * N / 2];
int root[N];
int find_root(int x)
{
	int temp = x;
	while(temp != root[temp])
		temp = root[temp];
	root[x] = temp;
	return temp;
}
int cmp(struct edge a, struct edge b)
{
	return a.len < b.len;
}
int main()
{
	int n, i, j, a, b, cnt, x, y, temp, ans;
	while(scanf("%d", &n) != EOF)
	{
		cnt = -1;
		ans = 0;

		for(i=1; i<=n; i++)
			root[i] = i;
		for(i=1; i<=n; i++)
			for(j=1; j<=n; j++)
			{
				scanf("%d", &temp);
				if(j > i)
				{
					e[++cnt].x = i;
					e[  cnt].y = j;
					e[  cnt].len = temp;
				}
			}
		sort(e, e + cnt + 1, cmp);
		for(i=0; i<=cnt; i++)
		{
			a = find_root(e[i].x);
			b = find_root(e[i].y);
			if(a != b)
			{

				ans += e[i].len;
				if(a > b)
					root[a] = b;
				else
					root[b] = a;
			}
		}
		printf("%d\n", ans);
	}
    return 0;
}


你可能感兴趣的:(POJ 1258 Agri-Net 【最小生成树】)