1. 程式人生 > >HDU 1233 還是暢通工程(最小生成樹prim 模板)

HDU 1233 還是暢通工程(最小生成樹prim 模板)

Problem Description

某省調查鄉村交通狀況,得到的統計表中列出了任意兩村莊間的距離。省政府“暢通工程”的目標是使全省任何兩個村莊間都可以實現公路交通(但不一定有直接的公路相連,只要能間接通過公路可達即可),並要求鋪設的公路總長度為最小。請計算最小的公路總長度。

 

 

Input

測試輸入包含若干測試用例。每個測試用例的第1行給出村莊數目N ( < 100 );隨後的N(N-1)/2行對應村莊間的距離,每行給出一對正整數,分別是兩個村莊的編號,以及此兩村莊間的距離。為簡單起見,村莊從1到N編號。
當N為0時,輸入結束,該用例不被處理。

 

 

Output

對每個測試用例,在1行裡輸出最小的公路總長度。

 

 

Sample Input

 

3 1 2 1 1 3 2 2 3 4 4 1 2 1 1 3 4 1 4 1 2 3 3 2 4 2 3 4 5 0

 

 

Sample Output

 

3 5

 

 

最小生成樹的模板題,可以用prim和kruaskal,這裡我用是prim

 

#include<iostream>
#include<cstdio>
#include<cstring>
#define inf 0x3f3f3f3f
using namespace std; //最小生成樹 
int map[105][105],vis[105],dis[105];
int n;
void init()
{
	for(int i=1;i<=n;i++)
	for(int j=1;j<=n;j++)
	map[i][j]=(i==j)?0:inf;
}
void prim()
{
	memset(dis,inf,sizeof(dis));
	memset(vis,0,sizeof(vis));
	int sum=0;
	int cnt=1;
	int i,j;
	vis[1]=1;
	for(i=1;i<=n;i++)
	dis[i]=map[1][i];
	for(j=1;j<=n;j++)
	{
		int minn=inf;
		int k=1;
	for(i=1;i<=n;i++)
	{
		
		if(!vis[i]&&dis[i]<minn) 
		{
			minn=dis[i];
			k=i;
		}
	}
		sum+=dis[k];
		vis[k]=1;
		
		for(i=1;i<=n;i++)
		{
			if(!vis[i]&&dis[i]>map[k][i])
			{
				dis[i]=map[k][i];
				//vis[i]=1;
			}
		}
	
}
		cout<<sum<<endl;  //輸出最小生成樹的權值和 
}
int main()
{
	while(scanf("%d",&n)&&n)
	{
		init(); //初始化map矩陣 
		int a,b,w,i,j;
		for(i=1;i<=n*(n-1)/2;i++)   //輸入矩陣 
		{
			scanf("%d%d%d",&a,&b,&w);
		 	if(w<map[a][b]) 
		 	map[a][b]=map[b][a]=w; 
		}
		prim();
	}
	return 0;
}#include<iostream>
#include<cstdio>
#include<cstring>
#define inf 0x3f3f3f3f
using namespace std; //最小生成樹 
int map[105][105],vis[105],dis[105];
int n;
void init()
{
	for(int i=1;i<=n;i++)
	for(int j=1;j<=n;j++)
	map[i][j]=(i==j)?0:inf;
}
void prim()
{
	memset(dis,inf,sizeof(dis));
	memset(vis,0,sizeof(vis));
	int sum=0;
	int cnt=1;
	int i,j;
	vis[1]=1;
	for(i=1;i<=n;i++)
	dis[i]=map[1][i];
	for(j=1;j<=n;j++)
	{
		int minn=inf;
		int k=1;
	for(i=1;i<=n;i++)
	{
		
		if(!vis[i]&&dis[i]<minn) 
		{
			minn=dis[i];
			k=i;
		}
	}
		sum+=dis[k];
		vis[k]=1;
		
		for(i=1;i<=n;i++)
		{
			if(!vis[i]&&dis[i]>map[k][i])
			{
				dis[i]=map[k][i];
				//vis[i]=1;
			}
		}
	
}
		cout<<sum<<endl;  //輸出最小生成樹的權值和 
}
int main()
{
	while(scanf("%d",&n)&&n)
	{
		init(); //初始化map矩陣 
		int a,b,w,i,j;
		for(i=1;i<=n*(n-1)/2;i++)   //輸入矩陣 
		{
			scanf("%d%d%d",&a,&b,&w);
		 	if(w<map[a][b]) 
		 	map[a][b]=map[b][a]=w; 
		}
		prim();
	}
	return 0;
}