最小生成树

模板例题 洛谷 P3366 【模板】最小生成树

传送门

给大家俩种算法的模板：Prim和Kruskal

Prim是以更新过的节点的连边找最小值，Kruskal是直接将边排序。

Prim

#include<bits/stdc++.h>
using namespace std;
struct edge
{
	int to,next,w;
}eg[2*2*100005];
int cnt = 1, head[5005],dis[5005],vis[5005];
void add(int u, int v, int w)
{
	eg[cnt].to = v;
	eg[cnt].w  = w;
	eg[cnt].next = head[u];
	head[u] = cnt++;
}
int main()
{
	int n,m,ans = 0;
	scanf("%d%d",&n,&m);
	memset(head,-1,sizeof(head));
	for(int i = 1; i <= m; i++)
	{
		int u,v,w;
		scanf("%d%d%d",&u,&v,&w);
		add(u,v,w);
		add(v,u,w);
	}
	dis[1] = 0;
	vis[1] = 1;
	for(int i = 2; i <= n; i++)
	{
		dis[i] = 123456789;
	}
	int tot = 1,now;
	for(int i = head[1];i != -1; i= eg[i].next)
	{
		dis[eg[i].to] = min(dis[eg[i].to],eg[i].w);
	}
	while(tot < n)
	{
		int minn = 123456789;
		for(int i = 1; i <= n; i++)
		{
			if(vis[i]==0 && minn > dis[i])
			{
				minn = dis[i];
				now = i;
			}			
		}
		tot++;
		vis[now] = 1;
		ans += minn;
		for(int i = head[now]; i!=-1; i = eg[i].next)
		{
			if(eg[i].w < dis[eg[i].to]&&vis[eg[i].to]==0)
			dis[eg[i].to] = eg[i].w;
		}
	}
	printf("%d",ans);
	return 0;
}

Kruskal

#include<bits/stdc++.h>
using namespace std;
int f[5005];
struct edge
{
	int to,next,w;
}eg[2*2*100005];
int  find(int x)
{
	if(x==f[x])
	{
		return f[x];
	}
	return f[x]=find(f[x]);
}
bool cmp(edge x,edge y)
{
	return x.w<y.w;
} 
int main()
{
	int n,m,ans = 0,a,b,cnt;
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)
    {
        f[i]=i;
    }
    	for(int i = 1; i <= m; i++)
	{
		scanf("%d%d%d",&eg[i].to,&eg[i].next,&eg[i].w);
	}
	sort(eg,eg+m,cmp);
	 for(int i=0;i<m;i++)
    {
        a=find(eg[i].to);
		b=find(eg[i].next);
        if(a==b)
		{
		continue;
		} 
        ans+=eg[i].w;
        f[b]=a;
        if(++cnt==n-1)
        {
            break;
        }		
       
    }
	printf("%d",ans);
	return 0;
}

希望对大家有用哦​​​​​​