多元哈弗曼编码问题

问题描述:在一个操场的四周摆放着n堆石子,现要将石子有次序地合并成一堆。规定每次至少选2堆至多选k堆石子合并成新的一堆,合并的费用为新的一堆石子数。计算出将n堆石子合并成一堆的最大总费用和最小总费用。


算法设计:对于给定的n堆石子,计算合并成一堆的最大总费用和最小总费用。
数据输入:文件的第1行有2个正整数n和k,表示有n堆石子,每次至少选2堆至多选k堆石子合并。第2行有n个数,分别表示每堆石子的个数。
输入示例:
7 3
45 13 12 16 9 5 22
输出示例:
593 199 

思路:用最大堆每次选两堆最大的,总的就是最大费用。用最小堆每次选k堆最小的,总的就是最小的。(不知道为什么)

#include<iostream>
#include<vector>
#include<queue>
using namespace std;
int main()
{
	int n, k;
	while (cin >> n >> k)
	{
		vector<int>v;
		v.push_back(0);
		priority_queue<int, vector<int>, less<int>>p;
		priority_queue<int, vector<int>, greater<int>>q;
		for (int i = 1; i <=n; i++)
		{
			int num;
			cin >> num;
			v.push_back(num);
			p.push(num);
			q.push(num);
		}
		
		int max_sum=0, top;
		while (p.size() > 1)
		{
			top = p.top();
			p.pop();
			top+=p.top();
			p.pop();
			max_sum += top;
			p.push(top);
		
		}
		int min_sum = 0;
		while (q.size() > k)
		{
			top = 0;
			for (int i = 0; i < k; i++)
			{
				top += q.top();
				q.pop();

			}
			q.push(top);
			min_sum += top;

		}
		while (q.size())
		{
			top = q.top();
			min_sum += top;
			q.pop();
		}
		cout << "maxs_sum: " << max_sum << "min_sum: " << min_sum << endl;
	}

	return 0;
}
/*
7 3
45 13 12 16 9 5 22

*/

 

©️2020 CSDN 皮肤主题: 技术黑板 设计师:CSDN官方博客 返回首页