Agri-Net POJ 1258

農夫約翰被選爲他的鎮長!他的競選承諾之一是將互聯網連接到該地區的所有農場。他當然需要你的幫助。
農民約翰爲他的農場訂購了高速連接,並將與其他農民共享他的連接。爲了最大限度地降低成本,他希望鋪設最少量的光纖,將他的農場與所有其他農場連接起來。
給出連接每對服務器場需要多少光纖的列表,您必須找到將它們全部連接在一起所需的最小光纖數量。每個場必須連接到其他某個場,這樣數據包才能從任何一個場流向任何其他場。
任何兩個農場之間的距離都不會超過10萬。

prim模板很簡單

#include<iostream>
#include<algorithm>
#include<cstring>

using namespace std;

const int MAXN=105;
const int INF=0x3f3f3f3f;

int n;
int map[MAXN][MAXN];
int dis[MAXN];
int vis[MAXN];

int ans;

void prim()
{
	ans=0;
	memset(dis,INF,sizeof(dis));
	for(int i=1;i<=n;i++)
		dis[i]=map[1][i];//起點 
	memset(vis,0,sizeof(vis));
	vis[1]=1;
	for(int i=2;i<=n;i++)
	{
		int pos=-1;
		for(int j=1;j<=n;j++)
		{
			if(!vis[j]&&(pos==-1||dis[pos]>dis[j]))
				pos=j;	
		}
		if(pos==-1)
			break;
		vis[pos]=1;
		ans+=dis[pos];
		for(int k=1;k<=n;k++)
			if(!vis[k])
				dis[k]=min(dis[k],map[pos][k]);
	}
}

int main()
{
	while(cin>>n)
	{
		for(int i=1;i<=n;i++)
			for(int j=1;j<=n;j++)
				cin>>map[i][j];
		prim();
		cout<<ans<<endl; 
	}
	return 0;
}

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章