求最小生成树现在我所学到的有两种算法:kruskal算法以及prim算法。
- kruskal算法
这个算法实际上就是加边法,每次选择最小的边加入集合(加入的边不能构成回路),直到把所有的节点都找到为止。
#include<iostream>
#include<cstring>
#include <algorithm>
using namespace std;
const int Maxsize=100;
struct Node
{
int from,to,weight;
};
int Find(int parent[],int nodes)//找到nodes的根节点
{
int f;
f=nodes;
if(parent[f]>-1)
f=parent[f];
return f;
}
bool cmp(Node a,Node b)//对Node进行自定义排序
{
return a.weight<b.weight;
}
int main()
{
Node *edge;
int rode[Maxsize][Maxsize],v,a;
cin>>v>>a;
edge=new Node[Maxsize];
int parent[v+10];
memset(parent,-1,sizeof(parent));
for(int i=0;i<v;i++)
{
for(int j=0;j<v;j++)
{
cin>>rode[i][j];
}
}
int s=-1;
for(int i=1;i<v;i++)//初始化看看有多少边
{
for(int j=0;j<i;j++)
{
if(rode[i][j]!=100&&rode[i][j]!=0)
{
++s;
edge[s].from=j+1;
edge[s].to=i+1;
edge[s].weight=rode[i][j];
}
}
}
sort(edge,edge+s+1,cmp);//排序
int couns=0;//统计已经遍历的节点
for(int i=0;i<=s;i++)
{
int begins=edge[i].from;
int enda=edge[i].to;
int m=Find(parent,begins);
int n=Find(parent,enda);
if(m!=n)
{
cout<<edge[i].from<<" "<<edge[i].to<<" ";
parent[n]=m;
couns++;
if(couns==v-1)
{
break;
}
}
}
}
- prim算法
代码如下:
#include<iostream>
#include<cstring>
#include <algorithm>
using namespace std;
const int Maxsize=100;
const int Max=100000;
int getmin(int lowcast[],int v,int visi[])//找到从始节点到其他节点最小的那个权值返回,但是终点不能为已经访问过的点
{
int minn=lowcast[1],mis=1;
for(int i=2;i<v;i++)
{
if(lowcast[i]<minn&&visi[i]==0)
{
minn=lowcast[i];
mis=i;
}
}
return mis;
}
void prim(int rode[Maxsize][Maxsize],int v)
{
int lowcast[Maxsize],visi[Maxsize];
memset(lowcast,0,sizeof(lowcast));
memset(visi,0,sizeof(visi));
visi[0]=1;
for(int i=0;i<v;i++)//初始化始节点到其他节点的权值
{
lowcast[i]=rode[0][i];
}
for(int i=1;i<v;i++)
{
int k=getmin(lowcast,v,visi);//找到最小的
visi[k]=1;//设置为已经遍历
cout<<lowcast[k]<<" ";
lowcast[k]=Max;
for(int j=1;j<v;j++)//以新加入的节点为中转站更新始节点到其他节点的权值
{
if(lowcast[j]>rode[k][j]&&visi[j]==0)
{
lowcast[j]=rode[k][j];
}
}
}
}
int main()
{
int rode[Maxsize][Maxsize],v,a;
cin>>v>>a;
for(int i=0;i<v;i++)
{
for(int j=0;j<v;j++)
{
cin>>rode[i][j];
}
}
for(int i=0;i<v;i++)
{
for(int j=0;j<v;j++)
{
if(rode[i][j]==100||rode[i][j]==0)
{
rode[i][j]=Max;
}
}
}
prim(rode,v);
}