kruskal算法
核心思想:
每次选择图中最小边权的边,如果边的两个端点不在同一个连通块中,则将这条边加入到最小生成树中。
伪代码:
int Kruskal()
{
初始化已选的边数num=0,权值之和ans=0;
初始化父结点为自身;
给边按照权值从小到大排序;
for(遍历所有的边)
{
找出这条边的两个端点结点;//用并查集的查找父结点功能实现
if(这条边的两个端点在同一个连通块中)
{
把边加入到树中;//即合并父结点
ans+=该边的权值;
num++;
if(num==结点数-1)//即已形成最小生成树
break;
}
}
if(num!=结点数-1) //说明图不连通
return -1;
return ans;
}
代码实现:
#include <bits/stdc++.h>
using namespace std;
const int MAXN=1000;
struct Edge
{
int u,v;
int w; //边权
};
bool cmp(Edge a,Edge b)
{
return a.w<b.w;
}
class EdgeGraph
{
public:
EdgeGraph(int v,int e);
~EdgeGraph(){}
int Kruskal();
private:
int vertexnum,edgenum;
Edge edge[MAXN];
int FindParent(int v);
int parent[MAXN];
};
EdgeGraph::EdgeGraph(int n,int e)
{
vertexnum=n;edgenum=e;
for(int i=0;i<e;i++)
{
cout<<"请输入第"<<i+1<<"条边的起点、终点、权值:";
cin>>edge[i].u>>edge[i].v>>edge[i].w;
}
for(int i=0;i<n;i++)
parent[i]=i;
}
int EdgeGraph::FindParent(int v)
{
int t=v;
while(v!=parent[v])
v=parent[v];
int z=t; //存v的根结点
//路径压缩
while(t!=v)
{
z=parent[t];
parent[t]=v;
t=z;
}
return v;
}
int EdgeGraph::Kruskal()
{
int num=0,ans=0;
sort(edge,edge+edgenum,cmp);
for(int i=0;i<edgenum;i++)
{
int p1=FindParent(edge[i].u);
int p2=FindParent(edge[i].v);
cout<<p1<<" "<<p2<<" ";
if(p1!=p2)
{
parent[p1]=p2;
ans+=edge[i].w;
cout<<"ans="<<ans<<" ";
num++;
cout<<"num="<<num<<endl;
}
else cout<<endl;
if(num==vertexnum-1)
break;
}
if(num!=vertexnum-1)
{
return -1;
}
return ans;
}
int main()
{
int n,e;
cin>>n>>e;
EdgeGraph Tree(n,e);
int ans=Tree.Kruskal();
if(ans==-1)cout<<"该图不连通";
else cout<<ans;
}
/*
6 10
0 1 4
0 4 1
0 5 2
1 2 1
1 5 3
2 3 6
2 5 5
3 4 5
3 5 4
4 5 3
*/
常用于稀疏图。