使用STL中的优先队列处理“合并果子”问题

最新推荐文章于 2022-11-30 11:03:50 发布

JunJie_1107

最新推荐文章于 2022-11-30 11:03:50 发布

阅读量256

点赞数

分类专栏：算法文章标签：算法队列数据结构

本文链接：https://blog.csdn.net/weixin_43869898/article/details/106451876

版权

算法专栏收录该内容

33 篇文章 0 订阅

订阅专栏

本文探讨了一种果子合并问题，旨在通过合理安排果子合并顺序，以最小化合并过程中消耗的体力总和。文章提供了一种有效的算法思路，即始终选择当前堆中质量最小的两堆果子进行合并，并使用C++优先队列实现该算法，最后通过数据测试验证了算法的有效性。

摘要由CSDN通过智能技术生成

问题描述：

有n堆果子，每堆果子质量已知，现在需要把这些果子合并成一堆，但是每次只能把两堆果子合并到一起，同时会消耗与两堆果子质量之和等值的体力。显然，在进行n-1次合并之后，就只剩下一堆了。为了尽可能节省体力，请设计出合并的次序方案，使得耗费的体力最少，并给出消耗的体力值。

例如：3堆果子，质量依次为1，2，3。则最少消耗的体力为：先合并1和2，消耗3体力；然后合并3和3，耗费6体力，总共耗费9体力。

解题思路：每次都找当前堆中拥有最小的质量的两堆果子，合并之。然后继续上一步操作，直到剩余果子只有一堆。

可使用C++中的优先队列来实现，每次都取小顶堆的首部元素，取两次，然后合并，累加求和。

代码实现如下：

#define _CRT_SECURE_NO_WARNINGS
#include <iostream>
#include <queue>
using namespace std;

priority_queue<long long, vector<long long>, greater<long long>> q;

int main()
{
	int n;
	long long temp;
	scanf("%d", &n);
	for (int i = 0; i < n; ++i)
	{
		scanf("%lld", &temp);
		q.push(temp);
	}

	long long t1, t2, ans = 0;
	while (q.size() > 1)
	{
		t1 = q.top();
		q.pop();

		t2 = q.top();
		q.pop();

		q.push(t1 + t2);
		ans += t1 + t2;
	}
	printf("%lld", ans);
	return 0;
}