[洛谷 P1546 最短网络 Agri-Net]&最小生成树学习笔记

最小生成树学习笔记

最小生成树，图论中一种极其基础的算法。
针对此算法，主要有两种解法：
$$Prim\Theta (n^2)$$
$$Kruskal\Theta (mlogm)$$

---

Prim

prim非常简单，就是直接将离加入点最近的未加入点直接加入。
代码：

#include<iostream>
#include<cstring>
using namespace std;
int a[200][200],m[200],b,ans;
bool v[200];
int main()
{
	cin>>b;
	for(int i=1;i<=b;i++)
	for(int j=1;j<=b;j++)
		cin>>a[i][j];
	memset(m,0x7f,sizeof(m));
	m[1]=0;
	for(int i=1;i<=b;i++)
	{
		int mi=0;
		for(int j=1;j<=b;j++) if(!v[j]&&m[j]<m[mi]) mi=j;
		v[mi]=1;
		for(int j=1;j<=b;j++) if(!v[j]&&a[mi][j]<m[j]) m[j]=a[mi][j];
	}
	for(int i=1;i<=b;i++) ans+=m[i];
	cout<<ans;
}

据说有一种方法可以用堆优化到$\Theta (mlogn)$的就是不知道怎么做。

---

Kruskal

这个算法更简单，开始将所有点划分为不同集合。
然后选择最短的边，判断两边是否属于同一集合。
如果不是，将它们并为同一集合，累加答案。
回到第二步。
在判断上，我们可以用并查集优化，选最短边可以排序。
代码

#include<bits/stdc++.h>
using namespace std;
int fa[10000],b,c,d,e,ans;
struct qq{
	int dis,r1,r2;
}a[200000];
bool s(qq w1,qq w2)
{
	return w1.dis<w2.dis;
}
int ffa(int i)
{
	if(fa[i]!=i) fa[i]=ffa(fa[i]);
	return fa[i];
}
void unionn(int x,int y)
{
	int fffa=ffa(x);
	int fb=ffa(y);
	if(fffa!=fb) fa[fffa]=fb;  
}
int main()
{
	for(int i=1;i<10000;i++) fa[i]=i;
	cin>>b;
	for(int i=1;i<=b;i++)
	for(int j=1;j<=b;j++)
	{
		cin>>c;
		if(c!=0)a[++d]=qq{c,i,j};
	}
	sort(a+1,a+d+1,s);
	for(int i=1; i<=d;i++)
	{
		if(ffa(a[i].r1)!=ffa(a[i].r2))
		{
			unionn(a[i].r1,a[i].r2);
			ans+=a[i].dis;
			e++;
		}
		if(e==b-1)break;
	}
	cout<<ans;
}