hdu 1233- kruskal+prime

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

这道题是我在杭电OJ上找的最小生成树的模板题，由于是比较重要的算法，应该要做到随手拿的出来的程度，故而记录了下来，也是为了方便自己忘了的时候能够快速找到代码学习。

下面的代码是 kruskal+优先队列 ：

#include<iostream>        //kruskal+优先队列 
#include<cstdio>
#include<algorithm>
#include<queue>
using namespace std;
struct node
{
	int u,v,w;
	friend bool operator < (node a,node b){
		return a.w>b.w;
	}	
};
int p[105];
priority_queue<node> q;
int find(int x)
{
	return p[x]==x?x:p[x]=find(p[x]);
}
int kruskal()
{
	int ans=0;
	node tmp;
	while(!q.empty())
	{
		tmp=q.top();
		q.pop();
		int f1=find(tmp.u);
		int f2=find(tmp.v);
		if(f1!=f2){
			p[f2]=f1;
			ans+=tmp.w;
		}
	}
	return ans;
}
int main()
{
	int n;
	while(scanf("%d",&n)==1&&n)
	{
		for(int i=1;i<=n;i++) p[i]=i;
		int a,b,c;
		node temp;
		for(int i=0;i<n*(n-1)/2;i++){
			scanf("%d %d %d",&a,&b,&c);
			temp.u=a;
			temp.v=b;
			temp.w=c;
			q.push(temp);
		}
		int d=kruskal();
		printf("%d\n",d);
	}
	return 0;
}

下面的代码 kruskal紫书写法

总的来说kruskal算法适用于稀疏图，本题由于是密集图，宜采用prime算法，时间开销方面会小一些。

#include<iostream>      //kruskal紫书写法 
#include<cstdio>
#include<algorithm>
using namespace std;
int u[10005],v[10005],w[10005];  //	第i条边的两端点以及这条边的权值 
int p[10005],r[10005];
int n;
bool cmp(int i,int j)   //间接排序
{
	return w[i]<w[j];
}
int find(int x)
{
	return p[x]==x?x:p[x]=find(p[x]);
}
int kruskal()
{
	int ans=0;
	int m=n*(n-1)/2;
	for(int i=1;i<=n;i++) p[i]=i;
	for(int i=1;i<=m;i++) r[i]=i;
	sort(r+1,r+m+1,cmp);
	for(int i=1;i<=m;i++){
		int e=r[i];
		int f1=find(u[e]);
		int f2=find(v[e]);
		if(f1!=f2){
			p[f1]=f2;
			ans+=w[e];
		}
	}
	return ans;
}
int main()
{
	while(scanf("%d",&n)==1&&n)
	{
		int a,b,c;
		for(int i=1;i<=n*(n-1)/2;i++){
			scanf("%d %d %d",&a,&b,&c);
			u[i]=a;
			v[i]=b;
			w[i]=c; 
		}
		int d=kruskal();
		printf("%d\n",d);
	}
	return 0;
}

下面的代码prime算法

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#define INF 0x3f3f3f3f
using namespace std;
int parent[10005];
int G[10005][10005];
int vis[10005];
int ans[10005];
int n;
int main()
{
	while(scanf("%d",&n)==1&&n)
	{
		int a,b,d;
		for(int i=1;i<=n*(n-1)/2;i++){
			scanf("%d %d %d",&a,&b,&d);
			G[a][b]=d;
			G[b][a]=d;
		}
		for(int i=1;i<=n;i++) G[i][i]=0;
		for(int i=1;i<=n;i++){
			ans[i]=G[1][i];
		}
		memset(vis,0,sizeof(vis));
		vis[1]=1;
		int cnt=1,sum=0,k;
		while(cnt<n){
			cnt++;
			int minn=INF;
			for(int i=1;i<=n;i++){
				if(minn>ans[i]&&!vis[i]){
					k=i;
					minn=ans[i];
				}
			}
			vis[k]=1;
			sum+=ans[k];
			for(int i=1;i<=n;i++){
				if(G[k][i]<ans[i]&&!vis[i]){
					ans[i]=G[k][i];
				}
			}
		}
		printf("%d\n",sum);
	}
	return 0;
}