1014 - Prim最小生成树 - 打井（usaco2009oct）

打井Watering Hole

描述

农民John 决定将水引入到他的n(1<=n<=300)个牧场。他准备通过挖若

干井，并在各块田中修筑水道来连通各块田地以供水。在第i 号田中挖一口井需要花费W_i(1<=W_i<=100,000)元。连接i 号田与j 号田需要P_ij (1 <= P_ij <= 100,000 , P_ji=P_ij)元。

请求出农民John 需要为连通整个牧场的每一块田地所需要的钱数。

输入

第1 行为一个整数n。

第2 到n+1 行每行一个整数，从上到下分别为W_1 到W_n。

第n+2 到2n+1 行为一个矩阵，表示需要的经费（P_ij）。

输出

只有一行，为一个整数，表示所需要的钱数。

样例输入

4
5
4
4
3
0 2 2 2
2 0 3 3
2 3 0 4
2 3 4 0

样例输出

9

 

分析

蛮基础的，稍微想一下，还是可做的

对于每口井，我们新建一个源点 S ，从S往每个井连一条权值为W_i的边，井与井之间就连权值为P_ij的边

由于每口井都需要有水，所以我们在新建的这个图上跑一遍最小生成树

忘了Prim+堆优化版本的怎么写了，就先贴一个普通版的

 

代码

#include<bits/stdc++.h>
#define N 305
using namespace std;
int n,w[N],p[N][N];
int d[N],vis[N];
int main(){
	scanf("%d",&n);
	int i,j,k;
	memset(p,127,sizeof(p));
	for(i=1;i<=n;++i){
		scanf("%d",&w[i]);
	} 
	for(i=1;i<=n;++i)
		for(j=1;j<=n;++j)
			scanf("%d",&p[i][j]);
	for(i=1;i<=n;++i)
		p[0][i]=w[i];
	memset(d,127,sizeof(d));
	memset(vis,0,sizeof(vis));
	d[0]=0;
	for(i=0;i<=n;++i){
		int k=n+1;
		for(j=0;j<=n;++j)
			if(!vis[j]&&(d[j]<d[k]))
				k=j;
		vis[k]=1;
		for(j=0;j<=n;++j)
			if(!vis[j]&&p[k][j]<d[j])
				d[j]=p[k][j];
	} 
	int mst=0;
	for(i=0;i<=n;++i)
	{
			mst+=d[i];
	}
	printf("%d",mst);
	return 0;
}