最小生成树总结
1.Prim
2.Kruskal
Prim
基于邻接矩阵的算法描述。
点到集合的最小距离!
此算法的关键在于建立两个集合U和V-U,U为已经选中的点集,V-U为还没纳入的点,在寻找的过程中在V-U中找点到集合U中权值最小的边,并将端点纳入集合,纳入集合后要考虑是否需要修改还没纳入集合U的点到集合U的距离。
①通过lowcost数组标识点是属于U还是V-U;lowcost[i]=0表示i点已纳入U,lowcost[i]!=0表示i点还没纳入。
②对于V-U中的顶点 j 到集合U有多条边,最小的位(k,j)则lowcost[j]=此权值,cloeset[j]=k;若没边,则lowcost[j]=+oo;
算法步骤
1.初始化lowcost数组,开始时即将起点v纳入集合U,并循环设置(1<=j<=n)设lowcost[j]=g.edges[v][j];
closest[j]=v;
2.(2<=i<=n);引入辅助值mincost,初始值为+oo,(2<=j<=n),循环找lowcost中最小值,并记录下标,用k保存
3.置lowcost[k]=0;判断是否需要更新lowcost;若lowcost[j]!=0&&lowcost[j]>g.edges[k][j];则更新lowcost[j]=g.eddges[k][j];closest[j]=k;
//基于邻接矩阵的Prim时间复杂度为O(n2);效率不很高,与边无关,适合稠密图
代码如下`
void Prim(int v){
int lowcost[n+10];
int closest[n+10];
int mincost;
int k;
for(int i=1;i<=n;i++){
lowcost[i]=Edges[v][j];
closest[i]=v;
}
for(int i=2;i<=n;i++){
mincost=9999999;
for(int j=2;j<=n;j++){
if(lowcost[j]!=0&&lowcost[j]<mincost){
mincost=lowcost[j];
k=j;
}
}
lowcost[k]=0;
for(int j=2;j<=n;j++){
if(lowcost[j]!=0&&lowcost[j]>edges[k][j]){
lowcost[j]=edges[k][j];
closest[j]=k;
}
}
}
}
此算法可做一点小优化,即不采用邻接矩阵存图,采用链式前向星结构存图。
struct Edge{
int to;
int w;
int next;
}
Edge edges[100];
int head[100];
int num=0;
void add(int u,int v,int w){
++num;
edges[num].to=v;
edges[num].w=w;
edges[num].next=head[u];
head[u]=num;
从而改进的Prim如下:`
void Prim(int u){
int lowcost[101];
int mincost;
int k=0;
int INF=999999;
for(int i=1;i<=n;i++){
lowcost[i]=INF;
}
lowcost[u]=0;
for(int i=head[u];i!=0;i=edges[i].next){
lowcost[edges[i].to]=min(lowcost[edges[i].to],edges[i].w);
}
for(int i=2;i<=n;i++){
mincost=INF;
for(int j=2;j<=n;j++){
if(lowcost[j]!=0&&lowcost[j]<mincost){
mincost=lowcost[j];
k=j;
}
}
lowcost[k]=0;
for(int j=head[k];j!=0;j=edges[j].next){
int v=edges[j].to;
if(lowcost[v]!=0&&lowcost[v]>edges[j].w){
lowcost[v]=edges[j].w;
}
}
}
这样可使速度快一点;后续学习到新的优化方法会继续更新。
Kruskal
此算法采用并查集判断选的边是否组成环路,同时采用边集数组存图
///并查集基本操作可查看我的博客
算法思想:
将边按从小到大排序,依次选取每一条边,若选的边没有与现在的边形成环路,则选择,否则就跳过。
步骤:
①初始化并查集,建立并查集,初始化计数器k,MST;
②排序边集数组。
③循环选取边
struct Edge{
int u;
int v;
int w;
bool operator<(const Edge& s)const{
return w<s.w;
}
}
Edge edges[100];
void Kruskal(){
int k=0;
int MST=0;
for(int i=1;i<=M;i++){
parent[i]=i;
}
sort(edges+1,edges+1+M);
for(int i=1;i<=M;i++){
if(Find(edges[i].u)!=Find(edges[i].v)){
Union(edges[i].u,edges[i].v);
k++;
}
if(k==n-1) break;
}
}