Kruskal算法(适用稀疏图)
主要思想:
将边进行排序,再依次从小到大添加边,同时用并查集判断是否出现环路,当添加的边数为顶点数-1时结束;
并查集:
int p[MAXN];
void p_csh(){ //初始化
for(int i=1;i<=n;i++) p[i]=i;
}
int getroot(int k){ // 找根
if(p[k]==k) return k;
else return getroot(p[k]);
}
void merge(int x,int y){ // 将两个不相同的集合合并
int xp = getroot(x),yp = getroot(y);
if(xp!=yp) p[xp] = yp;
}
例题:P3366 【模板】最小生成树
代码:
#include<bits/stdc++.h>
using namespace std;
int n,m,p[5005],ans;
void p_csh()
{
for(int i=1;i<=n;i++) p[i]=i;
}
int getroot(int k)
{
if(p[k]==k) return k;
else return getroot(p[k]);
}
void merge(int x,int y)
{
int xp = getroot(x),yp = getroot(y);
if(xp!=yp) p[xp] = yp;
}
struct edge{ //使用结构体存边的信息
int a,b,v;
}e[200005];
bool cmp(edge e1,edge e2)
{
return e1.v < e2.v;
}
int main()
{
cin>>n>>m;
for(int i=1;i<=m;i++)
cin>>e[i].a>>e[i].b>>e[i].v;
sort(e+1,e+m+1,cmp);
p_csh(); //初始化
int num=0;
for(int i=1;i<=m;i++) //从小到大遍历边
{
if(getroot(e[i].a)!=getroot(e[i].b)) // 判断两个顶点不在同一个集合
{
merge(e[i].a,e[i].b); // 合并两个集合
ans+=e[i].v; //最小生成树的边的和
num++; //已添加的边数
}
if(num == n-1) break; // 当边数 = 顶点数-1,树找完成
}
if(num == n-1) cout<<ans;
else cout<<"orz"; // 当边数 != 顶点数-1,图不连通,不能构成树
return 0;
}