bzoj 1601: [Usaco2008 Oct]灌水（最小生成树）

1601: [Usaco2008 Oct]灌水

Time Limit: 5 Sec   Memory Limit: 162 MB
Submit: 1978   Solved: 1297
[ Submit][ Status][ Discuss]

Description

Farmer John已经决定把水灌到他的n(1<=n<=300)块农田，农田被数字1到n标记。把一块土地进行灌水有两种方法，从其他农田饮水，或者这块土地建造水库。 建造一个水库需要花费wi(1<=wi<=100000),连接两块土地需要花费Pij(1<=pij<=100000,pij=pji,pii=0). 计算Farmer John所需的最少代价。

Input

*第一行：一个数n

*第二行到第n+1行：第i+1行含有一个数wi

*第n+2行到第2n+1行：第n+1+i行有n个被空格分开的数，第j个数代表pij。

Output

*第一行：一个单独的数代表最小代价.

Sample Input

4

5

4

4

3

0 2 2 2

2 0 3 3

2 3 0 4

2 3 4 0

Sample Output

9

模板水题- -

虚构一个点n+1，这个点与所有土地都连一条长度为建造水库所需花费的边

然后求最小生成树

#include<stdio.h>
#include<string.h>
#define inf 1044266558
int road[305][305], bet[305], flag[305];
int main(void)
{
	int n, i, x, temp, j, k, ans;
	while(scanf("%d", &n)!=EOF)
	{
		for(i=1;i<=n;i++)
		{
			scanf("%d", &x);
			road[i][n+1] = road[n+1][i] = x;
		}
		for(i=1;i<=n;i++)
		{
			for(j=1;j<=n;j++)
			{
				scanf("%d", &road[i][j]);
				if(i==j)
					road[i][j] = inf;
			}
		}
		ans = 0;
		memset(flag, 0, sizeof(flag));
		for(i=2;i<=n+1;i++)
			bet[i] = road[1][i];
		for(i=2;i<=n+1;i++)
		{
			k = inf;
			for(j=2;j<=n+1;j++)
			{
				if(flag[j]==0 && bet[j]<k)
				{
					k = bet[j];
					temp = j;
				}
			}
			ans += k;
			flag[temp] = 1;
			for(j=2;j<=n+1;j++)
			{
				if(flag[j]==0 && road[temp][j]<bet[j])
					bet[j] = road[temp][j];
			}
		}
		printf("%d\n", ans);
	}
	return 0;
}