prim最小生成树算法

const int maxv=1000;
const int inf=1000000000;

1、邻接矩阵版prim 

int G[maxv][maxv];
int n;
int d[maxv];
bool vis[maxv]={false};

int prim()
{
	fill(d,d+maxv,inf);
	d[0]=0;
	
	int ans=0;
	
	for(int i=0;i<n;i++)
	{
		int u=-1;
		int min=inf;
		for(int j=0;j<n;j++)
		{
			if(vis[j]==false&&d[j]<min)
			{
				u=j;
				min=d[j];
			}
		}
		if(u==-1) return -1;
		vis[u]==true;
		ans+=d[u];
		for(int v=0;v<n;v++)
		{
			if(vis[v]==false&&G[u][v]!=inf&&G[u][v]<d[v])
			{
				d[v]=G[u][v];
			}
		}
	}
	return ans;
}

2、邻接表版prim

struct Node
{
	int v;
	int dis;
};

vector<Node> adj[maxv];
int n;
int d[maxv];
bool vis[maxv]={false};

int prim()
{
	fill(d,d+maxv,inf);
	d[0]=0;
	
	int ans=0;
	
	for(int i=0;i<n;i++)
	{
		int u=-1;
		int min=inf;
		for(int j=0;j<n;j++)
		{
			if(vis[j]==false&&d[j]<min)
			{
				u=j;
				min=d[j];
			}
		}
		if(u==-1) return -1;
		vis[u]==true;
		ans+=d[j];
		for(int j=0;j<adj[u].size();j++)
		{
			int v=adj[u][j].v;
			if(vis[v]==false&&adj[u][j].dis<d[v])
			{
				d[v]=G[u][v];
			}
		}		
	}
	return ans;
}

3、测试代码

#include<cstdio>
#include<algorithm>
using namespace std;

const int maxv=1000;
const int inf=1000000000;

int n,m,G[maxv][maxv];
int d[maxv];
bool vis[maxv]={false};

int prim()
{
	fill(d,d+maxv,inf);
	d[0]=0;
	
	int ans=0;
	
	for(int i=0;i<n;i++)
	{
		int u=-1;
		int min=inf;
		for(int j=0;j<n;j++)
		{
			if(vis[j]==false&&d[j]<min)
			{
				u=j;
				min=d[j];
			}
		}
		if(u==-1) return -1;
		vis[u]=true;
		ans+=d[u];
		for(int v=0;v<n;v++)
		{
			if(vis[v]==false&&G[u][v]!=inf&&G[u][v]<d[v])
			{
				d[v]=G[u][v];
			}
		}
	}
	return ans;
}

int main()
{
	int u,v,w;
	scanf("%d%d",&n,&m);
	fill(G[0],G[0]+maxv*maxv,inf);
	for(int i=0;i<m;i++)
	{
		scanf("%d%d%d",&u,&v,&w);
		G[u][v]=G[v][u]=w;
	}
	int ans=prim();
	printf("%d\n",ans);
	return 0;
}