Boxes And Balls(三叉哈夫曼)

梦中醉卧惊起

于 2022-02-15 11:39:17 发布

阅读量232

点赞数 2

分类专栏：算法竞赛文章标签：算法

本文链接：https://blog.csdn.net/gpc_123/article/details/122941163

版权

算法竞赛专栏收录该内容

14 篇文章 0 订阅

订阅专栏

这篇博客探讨了三叉哈夫曼堆在解决特定问题中的应用，即如何通过拆分和合并操作最小化球的惩罚和。文章通过示例解释了如何将球从非空盒子中取出并分配到其他盒子中，以达到最小化惩罚和的目标。算法通过优先级队列实现，先拆分偶数个球，然后逐步合并最小的组合，确保每次操作减少的惩罚和最大化。最后，给出了AC代码实现这一过程。

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

三叉哈夫曼堆

如果是偶数则加上一个0,变成一个奇数
拆的时候是先拆大的,保留小的;如果不能一次拆完,则第一次等于种类数之和,保证留下的是最小的组合
合并的时候先合并最小的,能够一次合并的越多越好
方法一:

if (q.size() % 2 == 0) q.push(0);
    ans = 0;
while (q.size() > 1)
    {
        ans1 = q.top();
        q.pop();
        ans2 = q.top();
        q.pop();
        ans3 = q.top();
        q.pop();
        ans = ans + ans1 + ans2 + ans3;
        q.push(ans1 + ans2 + ans3);
    }

方法二:

if(n%2==0){
	ans1 = q.top();  q.pop();
	ans2 = q.top();  q.pop();
	q.push(ans1 + ans2);
	sum += ans1 + ans2;
}
while(q.size()>=3){
	ans1 = q.top();  q.pop();
	ans2 = q.top();  q.pop();
	ans3 = q.top();  q.pop();
	q.push(ans1 + ans2 + ans3);
	sum += ans1 + ans2 + ans3;
}

Boxes And Balls
有n个不同的盒子，其中第一个盒子中有n种不同颜色的球，n个不同的盒子中第i个盒子代表需要装下所有颜色为i的球，每次操作需要把任意一个非空盒子的球全部取出，取出球的个数为惩罚，可以分成k=2或k=3份放入k个盒子中，直到第i个盒子中全部装满第i种颜色的球，求每次取出球惩罚的和最小是多少
由样例2分析一次把球拿出来罚时为14，这是要把球分为2还是3份呢，为了下一次的罚时更少当然是选择3份，所以分别把球给第3和第4个盒子，一份放回空的盒子中，这时放回的球数为5，下一次再拿出来罚时为5，而这时所剩的盒子为第1个和第2个总数正好为5。这个过程反过来就是三叉的哈夫曼构成过程
AC代码

#include <iostream>
#include <algorithm>
#include <queue>
using namespace std;
typedef long long ll;
const int maxn = 2e5 + 5;
priority_queue<ll, vector<ll>, greater<ll> > q;
ll ans1, ans2, sum, ans3;
ll a[maxn];
int main(){
	ll n;
	cin >> n;
	for (int i = 0; i < n;i++){
		cin >> a[i];
		q.push(a[i]);
	}
	if(n%2==0){
		ans1 = q.top();  q.pop();
		ans2 = q.top();  q.pop();
		q.push(ans1 + ans2);
		sum += ans1 + ans2;
	}
	while(q.size()>=3){
		ans1 = q.top();  q.pop();
		ans2 = q.top();  q.pop();
		ans3 = q.top();  q.pop();
		q.push(ans1 + ans2 + ans3);
		sum += ans1 + ans2 + ans3;
	}
	cout << sum << endl;
	return 0;
}