最小生成树问题

Kruskal算法求解

#include<bits/stdc++.h>
using namespace std;
const int maxn=5005;
const int maxm=200005;
int n,m;
int x,y,z;
struct edge{
	int u,v,w;
};
edge bian[maxm];
int f[maxn];
int ans=0,flag=0,tot=0;
int find(int x)
{
	if(x==f[x]) return x;
	return f[x]=find(f[x]);
}
bool cmp(edge x,edge y)
{
	return x.w<y.w;
}
void kruskal()
{
	for(int i=1;i<=m;i++)
	{
		int u=find(bian[i].u);
		int v=find(bian[i].v);
		if(u!=v)
		{
			ans=ans+bian[i].w;
			tot++;
			f[u]=v;
		}
		if(tot==n-1){
			flag=1; break;
		}
	}
}
int main()
{
	cin>>n>>m;
	for(int i=1;i<=n;i++) f[i]=i;
	for(int i=1;i<=m;i++)
		cin>>bian[i].u>>bian[i].v>>bian[i].w;
	sort(bian+1,bian+m+1,cmp);
	kruskal();
	if(flag==0) cout<<"orz";
	else cout<<ans;
	return 0;
}

Prime算法

#include<bits/stdc++.h>
using namespace std;
#define inf 0x7ffffff
#define maxn 5005
#define maxm 200005
struct edge{
	int v,w,next;
}e[maxm*2];
int head[maxn],dis[maxn],cnt,n,m,tot,now=1,ans;
bool vis[maxn];
void add(int u,int v,int w)
{
	e[++cnt].v=v;
	e[cnt].w=w;
	e[cnt].next=head[u];
	head[u]=cnt;
}
void init()
{
	cin>>n>>m;
	int u,v,w;
	for(int i=1;i<=m;i++)
	{
		cin>>u>>v>>w;
		add(u,v,w); add(v,u,w);
	}
}
int prime()
{
	for(int i=2;i<=n;i++)
		dis[i]=inf;
	for(int i=head[1];i;i=e[i].next)
	{
		dis[e[i].v]=min(dis[e[i].v],e[i].w);
	}
	while(++tot<n)
	{
		int minn=inf;
		vis[now]=1;
		for(int i=1;i<=n;i++)
		{
			if(!vis[i]&&minn>dis[i])
			{
				minn=dis[i];
				now=i;
			}
		}
		ans+=minn;
		for(int i=head[now];i;i=e[i].next)
		{
			int v=e[i].v;
			if(dis[v]>e[i].w&&!vis[v])
				dis[v]=e[i].w;
		}
	}
	return ans;
}
int main()
{
	init();
	cout<<prime()<<"\n";
	return 0;
}