最小生成树 hdu 1233 （还是畅通工程）

还是畅通工程（最小生成树）

 

还是畅通工程

 

 

 

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

 

 

 

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

题意：给出任意两个村庄的距离，求将所有村庄连起来的最小公路长度。

 

题解：

1.Prim算法

 

<span style="font-size:18px;">#include<cstdio>
#include<cstring>
#include<algorithm>
#define INF 0x3f3f3f3f
using namespace std;

int n;
int vis[110];
int dis[110];
int edge[110][110];
int prim(){
	memset (vis,0,sizeof(vis));
	memset (dis,INF,sizeof(dis));
	dis[1]=0;
	int ans=0;
	for (int i=1;i<=n;i++){
		int sum=INF;
		int k=-1;
		for (int j=1;j<=n;j++){
			if (dis[j]<sum&&!vis[j]){
				k=j;sum=dis[j];
			}
			
		}
		vis[k]=1;ans+=sum;
		for (int j=1;j<=n;j++){
			if (!vis[j]&&edge[k][j]<dis[j]){
				dis[j]=edge[k][j];
			}
		}
	}
	return ans;
}
int main(){
	while (scanf ("%d",&n)&&n){
		int a,b,w;
		int m=n*(n-1)/2;
		
		for (int i=1;i<=m;i++){
			scanf ("%d %d %d",&a,&b,&w);
			edge[a][b]=w;
			edge[b][a]=w;//无向图，需要正反向加入
		}
		printf ("%d\n",prim());
	}
	return 0;
}</span>

2.Kruskal算法

 

<span style="font-size:18px;">#include<cstdio>
#include<algorithm>
using namespace std;
	int n;
	int fa[110];
struct node{
	int a,b,val;
}r[5050];//数组开小，TLE 

bool cmp(node a,node b){
	return a.val<b.val;
}

void init(){
	for (int i=1;i<=n;i++)
	fa[i]=i;
	
}

int find(int x){
	if (x!=fa[x])
	fa[x]=find(fa[x]);
	return fa[x];
}

void Union(int a,int b){
	a=find(a);
	b=find(b);
	if (a!=b)
	fa[a]=b; 
} 

int main(){
	while (scanf ("%d",&n)&&n){
		int s =0;
		int num = 0;
		int m=n*(n-1)/2;
		int aa,bb;
		init();
		for (int i=1;i<=m;i++){
			scanf ("%d %d %d",&r[i].a,&r[i].b,&r[i].val);
		}
		sort (r+1,r+m+1,cmp);
		for (int i=1;i<=m;i++){
			aa=find(r[i].a);
			bb=find(r[i].b);
			if (aa!=bb){
				Union(aa,bb);
				num++;
				s+=r[i].val;
			}
			if (num==n-1) break;		
		}
		printf ("%d\n",s);
	}
	return 0;
}</span>

 

 

 

 

//wa
#include<cstdio>
#include<cstring>
#include<algorithm>
#define M 5050
using namespace std;
int n,m;
int fa[110];
struct node{
	int v,u,val;
}eg,e[M];

bool cmp(node aa,node bb){
	return aa.val<bb.val;
}

void init(){
	for (int i=1;i<=n;i++){
		fa[i]=i;
	}
	
}

int find(int x){
	if (x!=fa[x])
	fa[x]=find(fa[x]);
	return fa[x]; 
}
int same(int xx,int yy){
	if (find(xx)!=find(yy))
	return 0;
	return 1;
}

void Union(int a,int b){
	a=find(a);
	b=find(b);
	if (a!=b)
	fa[a]=b;
}
int kruskal(){
	init() ;
	sort(e,e+n,cmp);
	int ans=0;
	for (int i=0;i<m;i++){
		eg=e[i];
		int fx=find(eg.u);
		int fy=find(eg.v);
		if (!same(fx,fy)){
			Union(fx,fy);
			ans+=eg.val;
		}
	}
	return ans;
	
}
int main(){
	while (scanf ("%d",&n)&&n){
		m=n*(n-1)/2;	
		for (int i=0;i<m;i++){
			scanf ("%d %d %d",&e[i].u,&e[i].v,&e[i].val);
		}
		printf ("%d\n",kruskal());
		
	}
	
	return 0;
}