还是畅通工程(2006浙江大学研究生上机复试题目[最小生成树] hdoj 1233)

还是畅通工程

点击打开链接                   

Time Limit: 4000/2000 MS (Java/Others)    Memory Limit: 65536/32768 K (Java/Others)
Total Submission(s): 32960    Accepted Submission(s): 14854

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


   
   

    
    

     
     Hint
    
    
Hint
   
   
   
    
Huge input, scanf is recommended.
  
  

 

题目来源:点击打开链接  点击打开链接 点击打开链接

思路:

     从题意中我们知道，已经修过路的只需要从未修路的村庄中找到一个离已经连通的图中最近的一个点，将其并入到连通图即可，由此可看是最小生成树

    可以从边考虑，即利用普利姆算法或者克努斯卡尔算法求最小生成树

    边比较多的时候普利姆算法比较高效，否则另一个较高效

普利姆算法:

    从点出发，适用于稠密网

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
using namespace std;
#define size 105
#define INF 0x7ffffff
int g[size][size];
int dis[size];
bool vis[size];
int n,m;
void init()
{
	for(int i=0;i<n;i++)
     for(int j=0;j<n;j++)
        g[i][j]=INF;	
}

void Prim(int s,int e)
{
  int ans=0;
  for(int i=1;i<=n;i++)
  {
  	dis[i]=g[s][i];
  	vis[i]=false;
  }
  vis[s]=true;
  dis[s]=0;
  for(int i=1;i<n;i++)
  {
  	int minnum=INF;
  	int temp=0;
  	for(int j=1;j<=n;j++)
  	  if(!vis[j]&&dis[j]<minnum)
  	  {
  	  	temp=j;
  	  	minnum=dis[j];
	  }
	  ans+=minnum;
	  vis[temp]=true;
	  for(int j=1;j<=n;j++)
	    if(!vis[j]&&g[temp][j]!=INF&&g[temp][j]<dis[j])
	      dis[j]=g[temp][j]; 
  }
  printf("%d\n",ans);
}
int main()
{
	while(scanf("%d",&n),n)
	{
		int m=n*(n-1)/2;
		for(int i=0;i<m;i++)
		{
			int s,e,v;
			scanf("%d%d%d",&s,&e,&v);
			g[s][e]=g[e][s]=v;
		}
		Prim(1,n);
	}
	return 0;
}

克努斯克尔算法：

       从边出发，适用于稀疏网

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
using namespace std; 
int pre[105];
struct node
{
 int p1;
 int p2;            
 int v;                  
} way[5000];
int m,n;

void init()
{
	memset(pre,0,sizeof(pre));
	memset(way,0,sizeof(way));
	for(int i=1;i<=n;i++)
	 pre[i]=i;
}

int cmp(node a,node b) 
{
  return a.v<b.v;
}

int find(int x)                   
{
 int ret=x;
 while(pre[ret]!=ret) 
  ret=pre[ret];
  int t=x,r;
  while(t!=ret)         
  {
   r=pre[t];   
   pre[t]=ret;
   t=r;
  }
  return ret;
}

int join(int x,int y)          
{
  int fx=find(x);
  int fy=find(y);
  if(fx!=fy)
  {
   if(fx<fy)
     pre[fy]=fx;
    else
     pre[fx]=fy; 
    return 1;
  }
   return 0;
}

void Kruskal()
{
  int ans=0,cnt=0;
  for(int i=0;i<m;i++)
  {
   if(cnt==n-1)
      break;
   if(join(way[i].p1,way[i].p2))
   {
   	cnt++;
	ans+=way[i].v;
   }
  }
   printf("%d\n",ans);
}

int main()
{  
  while(scanf("%d",&n),n)
  {
   m=n*(n-1)/2;
   init();
   for(int i=0;i<m;i++)
    scanf("%d%d%d",&way[i].p1,&way[i].p2,&way[i].v);
    
    sort(way,way+m,cmp);
    
     Kruskal();
  }
  return 0;
}