多元Huffman编码问题（贪心算法）

多元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 

using namespace std;

/*************************

    求最大的总费用，就是每次把最大的 两 堆石子相加

*************************/
long long findMax(int a[],int n,int k)
{
    long long maxx = 0;
    priority_queue q;      //这是STL库中的一个函数，<>里面是数据类型，默认按照从大到小的顺序排列数组，排序方式好像是大顶堆
    for(int i = 0; i < n; i++)
    {
        q.push(a[i]);
    }
    while(q.size() > 2)
    {
        long long sum = 0;
        for(int i = 0; i < 2; i++)
        {
            sum += q.top();
            q.pop();
        }
        maxx += sum;
        q.push(sum);
    }
    while(!q.empty())
    {
        maxx += q.top();
        q.pop();
    }
    return maxx;
}

long long findMin(int a[],int n, int k)
{
    long long minn = 0;
    // 这么写就是按照从小到大的顺序排列队列中的数据
    priority_queue,greater > q;
    for(int i = 0; i < n; i++)
    {
        q.push(a[i]);
    }
    int m = n;
    if(m > k)   //如果要算最小得分，要注意k堆石子的合并应该放在最后，即最后留出k-1个数来合并才能得到最小值
    {
        while(m > k)
        {
            m = m - k + 1;//m - (k-1)
        }
        //以上，m现在肯定小于k
        //要让前面的m个数刚好能合并成1堆，需要加上k-m个零，用零来填充
        for(int i = 0; i < k - m; i++)
            q.push(0);

    }
    while(q.size() > k)
    {
        long long sum = 0;
        for(int i = 0; i < k; i++)
        {

            sum += q.top();
            q.pop();
        }
        minn += sum;
        q.push(sum);
    }
    while(!q.empty())
    {
        minn += q.top();
        q.pop();
    }
    return minn;
}

int main()
{
    int n,k;
    int a[100009];
    cin>>n>>k;
    long long maxx;
    long long minn;
    for(int i = 0; i < n; i++)
        cin>>a[i];
    maxx = findMax(a,n,k);
    minn = findMin(a,n,k);
    cout<