基于并查集的kruskal算法
生活随笔
收集整理的這篇文章主要介紹了
基于并查集的kruskal算法
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
#include <iostream> //并查集的kruskal算法
using namespace std;
const int max_ve=1005,max_ed=15005;
int n,m,i; //n,m分別記錄頂點數和邊數
struct node
{
int par,ans;
}vertex[max_ve]; //頂點
struct Edge
{
int u,v,weigh;
}edge[max_ed]; //邊
int cmp(const void* a,const void* b)
{
return (*(Edge*)a).weigh-(*(Edge*)b).weigh;
}
int find_p(int j) //在處理兩個頂點是否在同個集合內要用到并查集
{
if(vertex[j].par!=j)
{
vertex[j].par=find_p(vertex[j].par);
}
return vertex[j].par;
}
bool union_set(int s,int t)
{
int os=find_p(s),ot=find_p(t);
if(os==ot)
return false;
if(vertex[os].ans>vertex[ot].ans)
vertex[ot].par=os;
else
vertex[os].par=ot;
if(vertex[os].ans==vertex[ot].ans)
vertex[ot].ans++;
return true;
}
int main()
{
int sum,ct;
scanf("%d%d",&n,&m);
for(i=1;i<=n;++i) //初始化,頂點從1到n
{
vertex[i].par=i;vertex[i].ans=0;
}
for(i=0;i<m;++i) //輸入邊的信息:起點,終點,權重
{
scanf("%d%d%d",&edge[i].u,&edge[i].v,&edge[i].weigh);
}
qsort(edge,m,sizeof(edge[0]),cmp);
sum=0;ct=0;
for(i=0;i<m&&ct<n-1;++i) //當ct==n-1說明有n-1條邊,即所有的頂點都已連接上,終止算法
{
if(union_set(edge[i].u,edge[i].v))
{
++ct; //邊數+1
sum+=edge[i].weigh;
}
}
printf("%d\n",sum); //sum為最小生成樹所有邊的總和
return 0;
}
using namespace std;
const int max_ve=1005,max_ed=15005;
int n,m,i; //n,m分別記錄頂點數和邊數
struct node
{
int par,ans;
}vertex[max_ve]; //頂點
struct Edge
{
int u,v,weigh;
}edge[max_ed]; //邊
int cmp(const void* a,const void* b)
{
return (*(Edge*)a).weigh-(*(Edge*)b).weigh;
}
int find_p(int j) //在處理兩個頂點是否在同個集合內要用到并查集
{
if(vertex[j].par!=j)
{
vertex[j].par=find_p(vertex[j].par);
}
return vertex[j].par;
}
bool union_set(int s,int t)
{
int os=find_p(s),ot=find_p(t);
if(os==ot)
return false;
if(vertex[os].ans>vertex[ot].ans)
vertex[ot].par=os;
else
vertex[os].par=ot;
if(vertex[os].ans==vertex[ot].ans)
vertex[ot].ans++;
return true;
}
int main()
{
int sum,ct;
scanf("%d%d",&n,&m);
for(i=1;i<=n;++i) //初始化,頂點從1到n
{
vertex[i].par=i;vertex[i].ans=0;
}
for(i=0;i<m;++i) //輸入邊的信息:起點,終點,權重
{
scanf("%d%d%d",&edge[i].u,&edge[i].v,&edge[i].weigh);
}
qsort(edge,m,sizeof(edge[0]),cmp);
sum=0;ct=0;
for(i=0;i<m&&ct<n-1;++i) //當ct==n-1說明有n-1條邊,即所有的頂點都已連接上,終止算法
{
if(union_set(edge[i].u,edge[i].v))
{
++ct; //邊數+1
sum+=edge[i].weigh;
}
}
printf("%d\n",sum); //sum為最小生成樹所有邊的總和
return 0;
}
轉載于:https://www.cnblogs.com/mjc467621163/archive/2011/07/17/2108518.html
總結
以上是生活随笔為你收集整理的基于并查集的kruskal算法的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 怎样和客户一起搞定需求
- 下一篇: AMD Zen4 X670E主板设计图曝