多元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
Hint
请注意数据范围是否可能爆 int。
Source
#include <iostream>
#include <bits/stdc++.h>
#include <queue>
bool cmp1(long long int a, long long int b)
{
return (a>b);
}
using namespace std;
int main()
{
int n,k;
cin>>n>>k;
priority_queue<long long int, vector<long long int>, greater<long long int> > a;
priority_queue<long long int> b;
for(int i = 1; i <= n; i++)
{
int number;
cin>>number;
b.push(number);
a.push(number);
}
long long int sum = 0;
long long int sum1 = 0;
while(b.size() != 1){
sum = 0;
for(int i = 1; i <= 2; i++)
{
sum+=b.top();
sum1+=b.top();
b.pop();
}
b.push(sum);
}
cout<<sum1<<" ";
sum = 0;
sum1 = 0;
while(a.size()%(k-1)!=1)
{
a.push(0);
}
while(a.size()!= 1){
sum = 0;
for(int i = 1; i <= k; i++)
{
sum+=a.top();
sum1+=a.top();
a.pop();
}
a.push(sum);
}
cout<<sum1;
return 0;
}