codevs【1063】合并果子（小根堆）

最新推荐文章于 2024-01-05 22:29:36 发布

yj8023xx

最新推荐文章于 2024-01-05 22:29:36 发布

阅读量211

点赞数

CC 4.0 BY-SA版权

分类专栏：程序设计文章标签：小根堆

本文链接：https://blog.youkuaiyun.com/qq_43498798/article/details/98451838

程序设计专栏收录该内容

115 篇文章

订阅专栏

本文介绍了一种使用堆（小根堆）和优先队列解决最小合并成本问题的方法，通过不断取出并合并堆中最小的两个元素，然后将结果重新加入堆中，最终求得所有操作的总成本。文章提供了完整的C++代码实现，展示了如何手动模拟堆的调整过程。

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

题目链接：http://codevs.cn/problem/1063/
分析
这道题考查的是堆，我们首先取出两个最小的数，合并然后再加入到堆中，并调整堆结构，因为取的是最小的数，所以采用小根堆，这道题也可以用优先队列来做，下面采用的是手动模拟堆的方式。

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
#define N 10000 + 10
int heap[N],cnt;
int n,ans;
//模拟小根堆
void push(int x)//向上调整
{
	heap[++cnt] = x;//先插入到尾部，然后再调整
	int i = cnt;
	while(i > 1 && heap[i] < heap[i >> 1]) swap(heap[i],heap[i >> 1]),i >>= 1;
}
void pop()
{
	heap[1] = heap[cnt--];
	int i = 1,j = 2;//向下调整
	while(j <= cnt) {
		if(j < cnt && heap[j + 1] < heap[j]) j |= 1;//j += 1
		if(heap[i] > heap[j]) {
			swap(heap[i],heap[j]);
			i = j,j <<= 1;
		} else break;
	}
}
int main()
{
	scanf("%d",&n);
	int x;
	for(int i = 0; i < n; i++) {
		scanf("%d",&x);
		push(x);
	}
	int t;
	while(--n) {
		t = heap[1];pop();
		t += heap[1];pop();
		ans += t;
		push(t);
	}
	printf("%d\n",ans);
	return 0;
}