heap+prim

本文介绍了一种使用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;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值