最小生成树主要有两个算法
1.Prim算法
通过一道裸模板题来讲述 题目链接
prim算法主要是将图分成树内和树外,树内的点已经形成边权最小的树,此时用堆来实现树内到树外最短的边连上的点就是此时树到该点的最小边,直接入堆即可
#include<bits/stdc++.h>
using namespace std;
int n,m;
struct ty1{
int to;
int l;
int next;
}edge[500009];
int cnt=0;
int head[6000];
void add(int x,int y,int z)
{
edge[++cnt].to=y;
edge[cnt].l=z;
edge[cnt].next=head[x];
head[x]=cnt;
}
struct ty2{
int x;
int len;
bool operator <(const ty2 &a)const{
return len>a.len;
}
}tmp;
long long dis[6000];
int vis[6000];
long long ans=0;
bool prim()
{
int i;
priority_queue<ty2>q;
vis[1]=1;
dis[1]=0;
int sum=1;
for(i=head[1];i!=-1;i=edge[i].next)
{
q.push({edge[i].to,edge[i].l});
}
while(q.size())
{
tmp=q.top();
q.pop();
if (vis[tmp.x])
continue;
vis[tmp.x]=1;
ans+=tmp.len;
sum++;
for(i=head[tmp.x];i!=-1;i=edge[i].next)
{
int y=edge[i].to;
if (vis[y])
continue;
q.push({y,edge[i].l});
}
}
if (sum>=n) //判断是否所有点都连上了
return false;
else return true;
}
int main ()
{
cin>>n>>m;
memset(head,-1,sizeof(head));
memset(dis,0x3f3f3f3f,sizeof(dis));
int i,x,y,z;
for(i=1;i<=m;i++)
{
cin>>x>>y>>z;
add(x,y,z);
add(y,x,z);
}
if (prim())
cout<<"orz"<<endl;
else {
cout<<ans<<endl;
}
}
2.Kruskal算法
通过这一道题来展示该算法 题目 链接
先连接好每一条边,将边从小到大排序,采用贪心思想来直接相连短边,通过并查集的思想来表示当前两点已相连
#include<bits/stdc++.h>
using namespace std;
struct ty{
int x;
int y;
int l;
}edge[300000];
int fa[300000];
int vis[300009];
bool cmp(ty a,ty b)
{
return a.l<b.l;
}
int find(int x)
{
if (fa[x]!=x)
fa[x]=find(fa[x]);
return fa[x];
}
int main ()
{
int a,b,i,j,x;
cin>>a>>b;
int cnt=1;
int sum=0;
for(i=1;i<=b;i++)
{
for(j=1;j<=b;j++)
{
cin>>x;
if (x==0||x>a)
continue;
sum++;
edge[cnt].x=i;
edge[cnt].y=j;
edge[cnt].l=x;
cnt=cnt+1;
}
}
for(i=1;i<=b;i++) //超级原点,能到所有物品,边权为a
{
edge[cnt].x=0;
edge[cnt].y=i;
edge[cnt].l=a;
cnt=cnt+1;
}
for(i=0;i<=b;i++) //初始化
fa[i]=i;
sort(edge+1,edge+1+sum,cmp);//排序
int fx,fy;
int dum=0;
long long ans=0;
for(i=1;i<cnt;i++)
{
fx=find(edge[i].x);
fy=find(edge[i].y);
if (fx==fy)
continue;
ans+=edge[i].l;
fa[fx]=fy; //连通起来
}
cout<<ans<<endl;
return 0;
}