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;
}


郑重声明:本站内容如果来自互联网及其他传播媒体,其版权均属原媒体及文章作者所有。转载目的在于传递更多信息及用于网络分享,并不代表本站赞同其观点和对其真实性负责,也不构成任何其他建议。