ZOJ 1586 QS Network MST prim水题

http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemId=586

题目大意：

QS是一种生物，要完成通信，需要设备，每个QS需要的设备的价格不同，并且，这种设备只能在两个QS之间用一次，也就是说，如果一个QS需要和3个QS通信的话，它就必须得买3个设备，同时，对方三个也必须买对应的适合自己的设备。同时，每两个QS之间是有距离的，要完成通信还需要网线，给出每两个QS之间的网线的价值。求一棵生成树，使得所需要的费用最少。数据范围：所有数据都在1000以内。

思路：

吃了好吃的继续~

好久没写prim了。来水一发。1A。

把邻接矩阵加上对应的接听器的值prim即可。

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int MAXN=1000+10;
const int INF=9999999;
int map[MAXN][MAXN],n,dis[MAXN],a[MAXN],vis[MAXN];
int prim()
{
	for(int i=1;i<=n;i++)
	{
		dis[i]=INF;
		vis[i]=0;
	}

	int cur=1;
	vis[cur]=1;
	dis[cur]=0;
	for(int i=1;i<=n;i++)
	{
		for(int j=1;j<=n;j++)
			if(!vis[j] && map[cur][j] < dis[j])
				dis[j]= map[cur][j] ;

		int mini=INF;
		for(int j=1;j<=n;j++)
			if(!vis[j] && mini > dis[j])
				mini=dis[cur=j];
		vis[cur]=true;
	}
	int sum=0;
	for(int i=1;i<=n;i++)
		sum+=dis[i];
	return sum;
}
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]);
				map[i][j]+=a[i]+a[j];
			}
		}

		int ans=prim();
		printf("%d\n",ans);

	}
	return 0;
}