记录编号 52394 评测结果 AAAAAAAAAA
题目名称 [USACO Oct08] 挖水井 最终得分 100
用户昵称 Gravatarcstdio 是否通过 通过
代码语言 C++ 运行时间 0.033 s
提交时间 2013-02-04 15:47:45 内存使用 0.66 MiB
显示代码纯文本
#include<iostream>
#include<cstdio>
#include<cmath>
#include<iomanip>
using namespace std;
const int SIZE=302;//顶点数
const int MAX=0xffffff;
int n;
int ans=0;
int c[SIZE][SIZE]={0};
int w[SIZE]={0};//某点挖井花费
void Prim(int s){
	//s是起点,n是顶点数,c是邻接矩阵,f是存放结果的数组
	//顶点下标是1~n,而非0~n-1
	bool used[SIZE]={0};
	int f[SIZE]={0};
	int i,sum,min=-1;
	for(i=1;i<=n;i++) f[i]=c[s][i];
	used[s]=1,sum=1;
	while(sum<n){
		min=-1;
		for(i=1;i<=n;i++){
			if(used[i]) continue;
			if(min==-1||f[i]<f[min]) min=i;
		}
		if(min==-1) break;
		sum++;
		used[min]=1;
		ans+=f[min];
		for(i=1;i<=n;i++){
			if(used[i]) continue;
			if(c[min][i]<f[i]) f[i]=c[min][i];
		}
	}
}
int main(){
	freopen("water.in","r",stdin);
	freopen("water.out","w",stdout);
	scanf("%d",&n);
	int i,j;
	for(i=1;i<=n;i++) scanf("%d",&w[i]);
	for(i=1;i<=n;i++){
		for(j=1;j<=n;j++) scanf("%d",&c[i][j]);
	}
	for(i=1;i<=n;i++) c[i][n+1]=c[n+1][i]=w[i];//所谓超级源点
	n++;
	Prim(n);
	printf("%d\n",ans);
	return 0;
}