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