小顶堆计算最小带权路径

#include <iostream>
#include <algorithm>
using namespace std;
const int maxn=30;
int n;
int heap[maxn];

void downAdjust(int low,int high){
	int i=low,j=i*2;
	while(j<=high){
		if(j+1<=high && heap[j]>heap[j+1])
			j=j+1;
		if(heap[i]>heap[j]){
			swap(heap[i],heap[j]);
			i=j;
			j=i*2;
		}
		else break;
	}
}

void upAdjust(int low,int high){
	int i=high,j=i/2;
	while(j>=low){
		if(heap[i]<heap[j]){
			swap(heap[i],heap[j]);
			i=j;
			j=i/2;
		}
		else break;
	}
}

void insert(int x){
	heap[++n]=x;
	upAdjust(1,n);
}

void creatHeap(){
	for(int i=n/2;i>=1;i--){
		downAdjust(i,n);
	}
}
void heapSort(){
	for(int i=n;i>1;i--){
		swap(heap[1],heap[i]);
		downAdjust(1,i-1);
	}
}
int popHeap(){
	swap(heap[1],heap[n--]);
	downAdjust(1,n);
	return heap[n+1];
}

int main(int argc, char** argv) {
	int ans=0;
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
		scanf("%d",&heap[i]);
	creatHeap();
	int N=n;
	for(int i=0;i<N-1;i++){
		int temp=0;
		temp+=popHeap();
		temp+=popHeap();
		insert(temp);
		ans+=temp;
	}
	printf("%d",ans);
	return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值