九度OJ题目：1017 还是畅通工程

Kruskal

#include <iostream>
#include <algorithm>
using namespace std;
struct Node
{
	int startPoint;
	int endPoint;
	int length;
} nodes[5000];
bool cmp(const Node A,const Node B)
{
	return A.length<B.length;
}
int father[100];
int findSet(int num)
{
	//其实就是获取到分组的最初始的点的信息;
	if(father[num]!=num)
		return findSet(father[num]);
	else		
		return father[num];
}
bool UnionSet(int x,int y)
{
	int tmpx=findSet(x);
	int tmpy=findSet(y);
	if(tmpy==tmpx)//因为两个已经是一个组了;
		return false;
	father[tmpy]=tmpx;//如果可以合并，这里只需要更新其前驱节点，因为他最终还是会找到之前的组;
	return true;


}
int main()
{
	int num;
	while(cin>>num&&num)
	{
		for(int i=1;i<=num;i++)
			father[i]=i;
		int tmp=num*(num-1)/2;
		for (int i=0;i<tmp;i++)
			cin>>nodes[i].startPoint>>nodes[i].endPoint>>nodes[i].length;
		sort(nodes,nodes+tmp,cmp);
		int result=0,count=0;
		for (int i=0;i<tmp;i++)
		{
			if(UnionSet(nodes[i].startPoint,nodes[i].endPoint))
			{
				result=result+nodes[i].length;
				count++;
				if(count+1==num)
					break;
			}
		}
		cout<<result<<endl;
	}
	return 0;
}

Prim 可以和Dijkstra对比，会发现惊人的相似，Dijkstra只是多维护了一个点到点的距离，而Prim只需要维护最短距离

#include <iostream>
#include <algorithm>
using namespace std;
int A[100][100],low[100];
bool visited[100];
int MaxInt=1232000;
int Prim(int num)
{
	int minDis,result=0;
	int ppre=1;
	for (int i=1;i<=num;i++)
	{
		low[i]=A[1][i];
	}
	visited[ppre]=true;
	for (int i=1;i<num;i++)
	{
		minDis=MaxInt;
		for (int j=1;j<=num;j++)//找到最小;
		{
			if(!visited[j]&&low[j]<minDis)
			{
				minDis=low[j];
				ppre=j;
			}
		}
		result=result+minDis;
		for (int j=1;j<=num;j++)
		{
			if(!visited[j]&&A[ppre][j]<low[j])
				low[j]=A[ppre][j];
		}
		visited[ppre]=true;
	}
	return result;
}
int main()
{
	int num;
	while(cin>>num&&num)
	{
		for(int i=1;i<=num;i++)
		{
			visited[i]=false;
			for (int j=1;j<=num;j++)
			{
				A[i][j]=MaxInt;
			}
		}
		int tmp=num*(num-1)/2,x,y,z;
		for (int i=0;i<tmp;i++)
		{
			cin>>x>>y>>z;
			A[y][x]=A[x][y]=z;
		}
		cout<<Prim(num)<<endl;
	}
	return 0;
}

http://ac.jobdu.com/problem.php?pid=1017