1760多元Huffman编码问题（贪心）

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

Output 

593 199

Hint

请注意数据范围是否可能爆 int。

#include <iostream>
#include <algorithm>
#include <string.h>
#include <vector>
#include <queue>

#define inf 0x3f3f3f3f

using namespace std;

priority_queue<int, vector<int>, greater<int> > q1;///greater是从小到大
priority_queue<int, vector<int>, less<int> >q2;///less是从大到小

int main()
{
    int n, k, i, x;
    long long sum1, sum2, sum;
    cin >> n >> k;
    for(i=0;i<n;i++)
    {
        cin >> x;
        q1.push(x);
        q2.push(x);
    }
    while(q1.size()%(k-1) != 1)
    {
        q1.push(0);
    }
    sum1 = 0;
    while(q1.size()!=1)
    {
        sum = 0;
        for(i=0;i<k;i++)
        {
            sum += q1.top();
            q1.pop();
        }
        sum1 += sum;
        q1.push(sum);
    }
    sum2 = 0;
    while(q2.size()!=1)
    {
        sum = 0;
        for(i=0;i<2;i++)
        {
            sum += q2.top();
            q2.pop();
        }
        sum2 += sum;
        q2.push(sum);
    }
    cout << sum2 << " " << sum1 << endl;
    return 0;
}