多元Huffman编码问题
Time Limit: 1000 ms Memory Limit: 65536 KiB
Problem Description
在一个操场的四周摆放着n堆石子。现要将石子有次序地合并成一堆。规定每次至少选2 堆最多选k堆石子合并成新的一堆,合并的费用为新的一堆的石子数。试设计一个算法,计算出将n堆石子合并成一堆的最大总费用和最小总费用。
对于给定n堆石子,计算合并成一堆的最大总费用和最小总费用。
Input
输入数据的第1 行有2 个正整数n和k(n≤100000,k≤10000),表示有n堆石子,每次至少选2 堆最多选k堆石子合并。第2 行有n个数(每个数均不超过 100),分别表示每堆石子的个数。
Output
将计算出的最大总费用和最小总费用输出,两个整数之间用空格分开。
Sample Input
7 3 45 13 12 16 9 5 22
Sample Output
593 199
#include <bits/stdc++.h>
using namespace std;
int main()
{
priority_queue <int> queue1;
priority_queue <int, vector <int>, greater <int> > queue2;
int n, k;
int x;
long long sum1, sum2;
scanf("%d%d", &n, &k);
for(int i = 1; i <= n; i++)
{
scanf("%d", &x);
queue1.push(x);
queue2.push(x);
}
sum1 = 0;
while(queue1.size() > 1)
{
int num1 = queue1.top();
queue1.pop();
int num2 = queue1.top();
queue1.pop();
queue1.push(num1 + num2);
sum1 += num1 + num2;
}
while(queue2.size() % (k - 1) != 1) queue2.push(0);
sum2 = 0;
while(queue2.size() > 1)
{
int num2;
int temp = 0;
for(int i = 1; i <= k; i++)
{
num2 = queue2.top();
queue2.pop();
temp += num2;
}
queue2.push(temp);
sum2 += temp;
}
printf("%lld %lld\n", sum1, sum2);
}