heap+prim

最新推荐文章于 2023-11-04 00:00:00 发布

拂晓

最新推荐文章于 2023-11-04 00:00:00 发布

阅读量647

点赞数

CC 4.0 BY-SA版权

分类专栏：生成树文章标签： insert

本文链接：https://blog.youkuaiyun.com/freedomcheng/article/details/7497288

生成树专栏收录该内容

3 篇文章

订阅专栏

本文介绍了一种使用Prim算法求解最小生成树(MST)的C++实现方法。通过建立顶点之间的边集合并利用优先队列进行高效选择，实现了最小生成树的构建过程。该算法适用于加权无向图，并确保最终生成的树包含所有顶点且总权重最小。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

#include<cstdio>
#include<cstring>
using namespace std;

struct node
{
int num;
int dis;
node(int n,int d):num(n),dis(d){}
node(){}
};

int n;

node h[500000];

void insert(node nd,int& len)
{
len++;
h[len]=nd;
int loc=len;
while(loc>1)
{
if(h[loc].dis<h[loc>>1].dis)
{
node nt=h[loc];
h[loc]=h[loc>>1];
h[loc>>1]=nt;

loc>>=1;
}
else
{
break;
}
}
}

void heapfi(int loc,int len)
{
int min=loc;
int left=(loc<<1);
int right=left+1;

if(left<=len&&h[left].dis<h[min].dis)
{
min=left;
}
if(right<=len&&h[right].dis<h[min].dis)
{
min=right;
}
if(min!=loc)
{
node nt=h[loc];
h[loc]=h[min];
h[min]=nt;

heapfi(min,len);
}
}

node get(int&len)
{
node ret=h[1];
h[1]=h[len];
len--;
heapfi(1,len);
return ret;
}

int head[102];
int next[20005];
int point[20005];
int weight[20005];

int prim()
{
int ret=0;
int len=0;
bool fl[102]={0};
int low[102];
memset(low,0x1f,sizeof(low));
low[1]=0;
insert(node(1,0),len);
int cnt=0;
while(cnt<n&&len>0)
{
node nt=get(len);
if(fl[nt.num])continue;
fl[nt.num]=1;
ret+=nt.dis;
cnt++;
for(int e=head[nt.num];e!=-1;e=next[e])
{
if(low[point[e]]>weight[e])
{
low[point[e]]=weight[e];
insert(node(point[e],weight[e]),len);
}
}
}
return ret;
}

int main()
{
while(scanf("%d",&n),n)
{
memset(head,-1,sizeof(head));
memset(point,-1,sizeof(point));
memset(next,-1,sizeof(next));
int ec=n*(n-1)/2;
for(int i=1;i<=ec;i++)
{
int u,v,w;
scanf("%d%d%d",&u,&v,&w);
next[i]=head[u];
point[i]=v;
weight[i]=w;
head[u]=i;

next[i+ec]=head[v];
point[i+ec]=u;
weight[i+ec]=w;
head[v]=i+ec;
}
printf("%d\n",prim());
}
return 0;
}