prim + kruskal 实现

prim:

算法思想：

void prim(int n,float **c)
{
T=∅;
S={1};
while(S!=V)
{
(i,j)是i∈S,j∈V-S的最小权边;
T=T∪{(i,j)};
S=S∪{j};
}
}

时间复杂度为：O（n^2）

#include <iostream>
#include <string.h>
using namespace std;
const int m=10000;
int visited[10];
int arr[20][20];
int low[20];//low[i]用于记录到i权值最小的边
int vertex,edge;
int sum=0;
void prim()
{
	int i,j;
	int pos;
	int temp;
	pos=1;
	visited[1]=1;
	low[1]=0;

	for(i=1;i<=vertex;i++)
	{
		if(i!=1)
		{
			low[i]=arr[1][i];
		}
	}
	for(i=1;i<vertex;i++)
	{
		temp=m;
		for(j=1;j<=vertex;j++)
		{
			if(visited[j]==0&&low[j]<temp)
			{
				temp=low[j];
				pos=j;
			}
		}
		cout<<low[pos]<<endl;
		sum +=low[pos];
		visited[pos]=1;

		for(j=1;j<=vertex;j++)
		{
			if(visited[j]==0&&arr[pos][j]<low[j])
				low[j]=arr[pos][j];
		}
	}
}
int main()
{
	int i,j,value;
	int x,y;
	cin>>vertex>>edge;
	memset(visited,0,sizeof(int)*10);
	memset(arr,m,sizeof(int)*400);
	memset(low,m,sizeof(int)*20);
	for(i=1;i<=edge;i++)
	{
		cin>>x>>y>>value;
		arr[x][y]=arr[y][x]=value;
	}
	sum=0;
	prim();
	/*for(j=1;j<=vertex;j++)
		cout<<low[j]<<" ";*/
	cout<<sum<<endl;
	return 0;
}

kruskal:

void kruskal(edge E[],int []V)
{
U=V;
TE={};
while(T中连通分量不为1)
{
寻找最短的边（u,v）;
if(u、v来自两个不同的连通分量)
{
TE=TE∪{(u,v)};
合并u,v使他们在同一个连通分量中;
E=E-{(u,v)};
}
}
}

时间复杂度为O（elge）

#include <iostream>
#include <string.h>
using namespace std;
int vertex,edge;
int u[20],v[20],w[20];
int p[100];
const int m=0x3f3f3f3f;
int sum=0;

int  findmin()
{
	int temp,i,po;
	temp=m;
	po=-1;
	for(i=1;i<=edge;i++)
	{
		if(w[i]<temp)
		{
			temp=w[i];
			po=i;
		}
	}

	return po;
}
int find(int x)
{
	return p[x]==x?x:p[x]=find(p[x]);
}
void kruskal()
{
	int i,k;
	for(i=1;i<vertex;i++)//v个顶点，最小生成树有v-1个边
	{
		int fu;
		int fv;
		while(1){
			k=findmin();
			if(k==-1)
			{
				cout<<"findmin error!";
					return ;
			}
			fu=find(u[k]);
			fv=find(v[k]);

			if(fu!=fv)
				break;

			else
			{
				w[k]=m;
			}
		}
		{
			cout<<w[k]<<" ";
			sum +=w[k];
			if(p[fu]>p[fv])
				p[fv]=fu;
			else
				p[fu]=fv;

		}
		w[k]=m;
	}
}
void init()
{
	memset(u,0,sizeof(int)*20);
	memset(v,0,sizeof(int)*20);
	memset(w,0,sizeof(int)*20);
	memset(p,0,sizeof(int)*100);
}
int main()
{
	int i;
	cin>>vertex>>edge;

	init();

	for(i=1;i<=vertex;i++)
		p[i]=i;

	for(i=1;i<=edge;i++)
	{
		cin>>u[i]>>v[i]>>w[i];
	}

	sum=0;
	kruskal();
	cout<<endl<<sum<<endl;
	return 0;
}

<pre code_snippet_id="304642" snippet_file_name="blog_20140421_2_6578944">