hdu 1863 畅通工程 最小生成树模板入门题 prim+kruskal两种算法AC。

畅通工程

Time Limit: 1000/1000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 18866    Accepted Submission(s): 8012

Problem Description

省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通（但不一定有直接的公路相连，只要能间接通过公路可达即可）。经过调查评估，得到的统计表中列出了有可能建设公路的若干条道路的成本。现请你编写程序，计算出全省畅通需要的最低成本。

 

Input

测试输入包含若干测试用例。每个测试用例的第1行给出评估的道路条数 N、村庄数目M ( < 100 )；随后的 N
行对应村庄间道路的成本，每行给出一对正整数，分别是两个村庄的编号，以及此两村庄间道路的成本（也是正整数）。为简单起见，村庄从1到M编号。当N为0时，全部输入结束，相应的结果不要输出。

 

Output

对每个测试用例，在1行里输出全省畅通需要的最低成本。若统计数据不足以保证畅通，则输出“?”。

 

Sample Input

  
  

   
   3 3
1 2 1
1 3 2
2 3 4
1 3
2 3 2
0 100
  
  

 

Sample Output

  
  

   
   3
?
  
  

 

kruskal算法代码：

#include <cstdio>
#include <algorithm>
#include <cstring>
#define MAX 110
#define INF 1000000000
using namespace std ;

struct Edge{
	int x,y,w;
}edge[MAX];
int f[MAX] ;

bool cmp(const Edge &a ,const Edge &b)
{
	return a.w<b.w ;
}

void init()
{
	for(int i = 0 ; i < MAX; ++i)
	{
		f[i] = i ; 
	}
}
int find(int x)
{
	int r = x ;
	while(r != f[r])
	{
		r = f[r] ;
	}
	int temp ;
	while(x != r)
	{
		temp = f[x] ;
		f[x] = r ;
		x = temp ;
	}
	return r ;
}

int kruskal(int n , int m)
{
	int lowcost[MAX];
	bool visited[MAX] ;
	memset(visited,false,sizeof(visited)) ;
	sort(edge,edge+n,cmp);
	for(int i = 0 ; i <= m ; ++i)
	{
		lowcost[i] = INF ;
	}
	int index = 0 ;
	init() ;
	for(int i = 0 ; i < n ; ++i)
	{
		int fx = find(edge[i].x) , fy = find(edge[i].y) ;
		if(fx != fy)
		{
			lowcost[index++] = edge[i].w ;
			f[fx] = fy ;
		}
	}
	int sum = 0 ;
	for(int i = 0 ; i < m-1 ; ++i)
	{
		if(lowcost[i] == INF)
		{
			return INF ;
		}
		else
		{
			sum += lowcost[i] ;
		}
	}
	return sum ;
}

int main()
{
	int n , m ;
	while(~scanf("%d%d",&n,&m) && n)
	{
		for(int i = 0 ; i < n ; ++i)
		{
			int x , y , w ;
			scanf("%d%d%d",&x,&y,&w) ;
			edge[i].x = x ;
			edge[i].y = y ;
			edge[i].w = w ;
		}
		int sum = kruskal(n,m) ;
		if(sum == INF)
		{
			puts("?");
		}
		else
		{
			printf("%d\n",sum) ;
		}
	}
	return 0 ;
}

prim算法代码：

#include <stdio.h>
#include <string.h>
#define MAX 110
#define INF 1000000000

int graph[MAX][MAX] ,visited[MAX] ;

int prim(int m)
{
	int lowcost[MAX] , closest[MAX];
	for(int i = 1 ; i <= m ; ++i)
	{
		closest[i] = 1 ;
		lowcost[i] = graph[1][i] ;
	}
	lowcost[1] = 0 ;
	memset(visited,false,sizeof(visited)) ;
	visited[1] = true ; 
	for(int i = 0 ; i < m-1 ; ++i)
	{
		int min = INF , index=0 ;
		for(int j = 1 ; j <= m ; ++j)
		{
			if(!visited[j] && lowcost[j]<min)
			{
				min = lowcost[j] ;
				index=j ;
			}
		}
		visited[index] = true ;
		for(int j = 1 ; j <= m ; ++j)
		{
			if(!visited[j] && lowcost[j]>graph[index][j])
			{
				lowcost[j] = graph[index][j] ;
				closest[j] = index ;
			}
		}
	}
	int sum = 0 ;
	for(int i = 1 ; i <= m ; ++i)
	{
		if(lowcost[i] == INF)
		{
			return INF ;
		}
		sum += lowcost[i] ;
	}
	return sum ;
}

int main()
{
	int n , m ;
	while(~scanf("%d%d",&n,&m) && n)
	{
		for(int i = 0 ; i <= m ; ++i)
		{
			for(int j = 0 ; j <= i ; ++j)
			{
				graph[i][j] = graph[j][i] = INF ;
			}
		}
		for(int i = 0 ; i < n ; ++i)
		{
			int x , y , w ;
			scanf("%d%d%d",&x,&y,&w);
			graph[x][y] = graph[y][x] = w ;
		}
		int sum = prim(m) ;
		if(sum == INF)
		{
			puts("?");
		}
		else
		{
			printf("%d\n",sum) ;
		}
	}
	return 0 ;
}