记录编号 226768 评测结果 AAAAAAAAAA
题目名称 最优布线问题 最终得分 100
用户昵称 GravatarAntiLeaf 是否通过 通过
代码语言 C++ 运行时间 0.899 s
提交时间 2016-02-18 11:04:48 内存使用 9.00 MiB
显示代码纯文本
#include<cstdio>
#include<cstring>
#define INF 0x7fffffff
using namespace std;
const int maxn=1510;
int graph[maxn][maxn];
int lowcost[maxn],closeset[maxn],v;
bool used[maxn];
int prim(); 
int main(){
#define COGS
#ifdef COGS
	freopen("wire.in","r",stdin);
	freopen("wire.out","w",stdout);
#endif
	scanf("%d",&v);
	for(int i=0;i<v;i++)for(int j=0;j<v;j++)scanf("%d",&graph[i][j]);
	printf("%d",prim());
	return 0;
}
int prim(){
	int min,mst=0;
	for(int i=0;i<v;i++){
		lowcost[i]=graph[0][i];
		closeset[i]=0;
		used[i]=false;
	}
	used[0]=true;
	for(int i=1;i<v;i++){
		min=INF;
		int j=0;
		for(int k=1;k<v;k++){
			if(!used[k]&&lowcost[k]<min)min=lowcost[k],j=k;
		}
		used[j]=1;
		mst+=min;
		for(int k=1;k<v;k++){
			if(!used[k]&&graph[j][k]<lowcost[k]){
				lowcost[k]=graph[j][k];
				closeset[k]=j;
			}
		}
	}
	return mst;
}