hdoj 1233 还是畅通工程---最小生成树---prime算法

题目：http://acm.hdu.edu.cn/showproblem.php?pid=1233

可以用Kruskal来做，不过当图的边比较稠密的时候用prime会更快一些。

AC代码：296MS

 

#include <iostream>

#include <cstring>

#include <cstdio>

using namespace std;

const int inf=0xffffff;

int dis[105],sum,n;

int weight[105][105];

void prime()

{

	bool visit[105];

	int pre[105];

	memset(visit,0,sizeof(visit));

	for(int i=1;i<=n;i++)

		{dis[i]=weight[1][i];pre[i]=1;}

	visit[1]=1;

	int index,mmin;

	for(int i=1;i<=n-1;i++)

	{

		mmin=inf;

		for(int j=1;j<=n;j++)

		{

			if(!visit[j] && dis[j]<mmin)

			{

				mmin=dis[j];

				index=j;

			}

		}

		visit[index]=1;

		sum+=weight[pre[index]][index];

		for(int i=1;i<=n;i++)

		{

			if(!visit[i] && weight[index][i]<dis[i])

			{

				dis[i]=weight[index][i];

				pre[i]=index;

			}

		}

	}	

}

int main()

{



	while(scanf("%d",&n))

	{

		if(n==0)break; 

		int m=n*(n-1)/2;

		for(int i=1;i<=n;i++)

			for(int j=1;j<=n;j++)

				weight[i][j]=inf;

		for(int i=0;i<m;i++)

		{

			int x,y,z;

			scanf("%d %d %d",&x,&y,&z);

			weight[x][y]=z;

			weight[y][x]=z;

		}

		sum=0;

		prime();

		cout<<sum<<endl;

	}

}