loj 最小生成树（模板）

最新推荐文章于 2022-02-22 23:38:57 发布

touso

最新推荐文章于 2022-02-22 23:38:57 发布

阅读量107

点赞数

分类专栏：图论文章标签：最小生成树图论

本文链接：https://blog.csdn.net/touso/article/details/108993276

版权

图论专栏收录该内容

2 篇文章 0 订阅

订阅专栏

本文提供了最小生成树问题的模板代码，包括Kruskal和Prim两种算法。请注意，这些代码可能未经过完整测试，需要进一步调整以确保正确性。

摘要由CSDN通过智能技术生成

最小生成树
模板如下
kruskal算法

#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
const int maxn=2e5+10;
typedef long long ll;
ll ans=0;
int f[maxn],n,m;
struct node
{
	int x,y,z;
}s[maxn*4];
bool cmp(node a,node b)
{
	return a.z<b.z;
}
int find(int x)
{
	if(x==f[x])
	return x;
	else
	{
		f[x]=find(f[x]);
		return f[x];
	}
}
void kruskal()
{
	int f1,f2,k;
	k=0,ans=0;
	for(int i=1;i<=n;i++)
	f[i]=i;
	for(int i=1;i<=m;i++)
	{
		f1=find(s[i].x);
		f2=find(s[i].y);
		if(f1!=f2)
		{
			ans=ans+s[i].z;
			f[f1]=f2;
			k++;
			if(k==n-1)
			break;
		}
	}
	printf("%lld\n",ans);
}
int main()
{
	scanf("%d%d",&n,&m);
	for(int i=1;i<=m;i++)
	scanf("%d%d%d",&s[i].x,&s[i].y,&s[i].z);
	sort(s+1,s+1+m,cmp);
	kruskal();
	return 0;
 }

prim算法
这个代码并没有通过，只是个模板，将e数组换成结构体应该就可以了（我没有试）

#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
const int maxn=2e5+10;
int book[maxn];
typedef long long ll;
const int inf=0x3f3f3f3f;
ll dis[maxn],e[1010][1010],ans=0;
int n,m;
void prim(int x)
{
	memset(book,0,sizeof book);
	memset(dis,inf,sizeof dis);
	for(int i=0;i<=n;i++)
	dis[i]=e[x][i];
	book[x]=1;
	for(int i=1;i<n;i++)
	{
		int mi=inf,p;
		for(int j=1;j<=n;j++)
		{
			if(!book[j]&&dis[j]<mi)
			{
				mi=dis[j];
				p=j;
			}
		}
		book[p]=1;
		ans+=dis[p];
		for(int k=1;k<=n;k++)
		if(!book[k]&&dis[k]>e[p][k])
		dis[k]=e[p][k];
	}
}
int main()
{
	
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)
	{
		for(int j=1;j<=n;j++)
		{
			if(i==j)
			e[i][j]=0;
			else
			e[i][j]=inf;
		}
	}
	for(int i=0;i<m;i++)
	{
		int x,y,val;
		scanf("%d%d%d",&x,&y,&val);
		if(e[x][y]>val)
		e[x][y]=e[y][x]=val;
	}
	prim(1);
	printf("%lld\n",ans);
	return 0;
 }