ZOJ-1586-QS Network

用Prime的 最小生成树题

代码：

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
const int maxn=1001;
const int inf=1<<28;
using namespace std;
int n;
int a[maxn],map[maxn][maxn],dist[maxn];
int Prime()
{
    int ans=0;
    for(int i=1;i<=n;i++)
	dist[i]=map[1][i];
    dist[1]=-1;
    for(int i=2;i<=n;i++)
    {
	int mini=inf,pos=-1;
	for(int j=1;j<=n;j++)
	    if(dist[j]!=-1&&dist[j]<mini)
	    {
		mini=dist[j];
		pos=j;
	    }
	ans+=mini;
	for(int j=1;j<=n;j++)
	    dist[j]=min(dist[j],map[pos][j]);
	dist[pos]=-1;
    }
    return ans;
}
int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
	    scanf("%d",&a[i]);
	for(int i=1;i<=n;i++)
	    for(int j=1;j<=n;j++)
	    {
		scanf("%d",&map[i][j]);
		if(i==j)
		    map[i][j]=inf;
		else
		    map[i][j]+=a[i]+a[j];
	    }
	int ans=Prime();
	printf("%d\n",ans);
    }
    return 0;
}