最小生成树(模板)
这里只介绍Kruskal算法,在NOIP考试中,还未成出现prime/Kruskal算法卡其中一种的情况
Kruskal算法基本思路:
首先把所有的边按照权值先从小到大排列,接着按照顺序选取每一条边,如果这条边的两个端点不属于同一集合,那么就将它们合并,直到所有的点都属于同一个集合为止。
算法实现:用一个结构体数组e来保存信息,再用sort排序,这里需要自己写一个函数cmp来定义排序标准,n个点只需要n-1条边,加边的时候用sum记录一共加了几条边,当sum==n-1时,退出循环
注意:这里需要用到并查集的知识,如果不懂并查集的读者,请自行百度,以后我有时间会贴上并查集的模板的……
这里先贴上几道基础的最小生成树的题:
城市公交网建设问题:
点击打开链接
局域网:点击打开链接
最优布线问题:
点击打开链接
OK,上代码:
#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
struct edge{int u,v,w;}e[1000];//起点,终点,权值
int n,m,t1,t2;
int f[1000],sum,tot;
bool cmp(edge a,edge b){return a.w<b.w;}
int getf(int v)
{
return f[v]==v?v:f[v]=getf(f[v]);//并查集核心。。。
}
int merge(int v,int u)
{
t1=getf(v);t2=getf(u);
if(t1!=t2){f[t2]=t1;return 1;}//先判断两点是否联通
else return 0;
}
int main()
{
cin>>n>>m;sum=0;
for(int i=1;i<=m;++i)
cin>>e[i].u>>e[i].v>>e[i].w;
sort(e+1,e+m+1,cmp);//排序,cmp函数见上面
for(int i=1;i<=n;++i)
f[i]=i;//初始化f数组
for(int i=1;i<=m;++i){
if(merge(e[i].u,e[i].v))
sum++,tot=tot+e[i].w;
if(sum==n-1) break;
}
cout<<tot<<endl;
return 0;
}
END