【模板】最小生成树

【模板】最小生成树

题目描述

如题，给出一个无向图，求出最小生成树，如果该图不连通，则输出orz

输入格式

第一行包含两个整数N、M，表示该图共有N个结点和M条无向边。（N<=5000，M<=200000）

接下来M行每行包含三个整数Xi、Yi、Zi，表示有一条长度为Zi的无向边连接结点Xi、Yi

输出格式

输出包含一个数，即最小生成树的各边的长度之和；如果该图不连通则输出orz

---

prim算法模板：

#include <iostream>
#define inf 10000000
using namespace std;
int map[5005][5005];
int dis[5005];
bool vis[5005];
int main()
{
	int n,m;
	cin>>n>>m;
	for(int i=1;i<=n;i++)
	for(int j=1;j<=n;j++) map[i][j]=inf;
	for(int i=1;i<=m;i++)
	{
		int x,y,z;
		cin>>x>>y>>z;
		if(z<map[x][y])map[x][y]=z;//两个点之间可能有多条边
		if(z<map[y][x])map[y][x]=z;
	}
	for(int i=1;i<=n;i++) dis[i]=map[1][i];//求每个点与点1的距离
	int count=1;//已经加入一个点
	vis[1]=true;
	int ans=0;
	while(count<n)
	{
	    int min=1;
	    for(int i=1;i<=n;i++) 
	    if(dis[i]<dis[min]&&!vis[i]) min=i;//寻找与目前已相连的部分距离最近的点
	    if(min==1) {cout<<"orz"<<endl;return 0;}//没有找到则不存在最短路
	    vis[min]=true;
	    count++;
	    ans+=dis[min];
	    for(int i=1;i<=n;i++) 
	    if(map[min][i]<dis[i]&&!vis[i])
	    dis[i]=map[min][i];//更新每个点到当前部分的最短距离
	}
	cout<<ans<<endl;
	return 0;
}

kruskal算法

#include <iostream>
#include <algorithm>
using namespace std;
int map[5005][5005];
struct node
{
	int begin;
	int end;
	int dis;
}a[200010];
bool cmp(node a,node b)
{
	return a.dis<b.dis;
}
int f[5005];
int get_root(int a)
{
	if(f[a]==a) return a;
	return f[a]=get_root(f[a]);
}
void marge(int a,int b)
{
	int n=get_root(a);
	int m=get_root(b);
	f[n]=m;
}
int main()
{
	int n,m;
	cin>>n>>m;
	for(int i=1;i<=n;i++) f[i]=i;
	for(int i=1;i<=m;i++)
	{
		cin>>a[i].begin>>a[i].end>>a[i].dis;
	}
	sort(a+1,a+1+m,cmp);
	int ans=0,count=0;
	for(int i=1;i<=m;i++)
	{
		if(get_root(a[i].begin)==get_root(a[i].end)) continue;
		marge(a[i].begin,a[i].end);
		count++;
		ans+=a[i].dis;
		if(count==n-1) {cout<<ans<<endl;return 0;}
	}
	cout<<"orz"<<endl;
	return 0;
}