多元Huffman编码问题

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

 要想费用最大，合并次数就要多，所以一次只合并2堆，且先合并大的；费用最小，合并次数就要少，所以一次合并k堆，且先合并小的。利用C++ STL里的优先队列即堆结构完成

#include <bits/stdc++.h>

using namespace std;


int main()
{
    int n, k;
    scanf("%d%d", &n, &k);
    priority_queue<long long> big;
    priority_queue<long long, vector<long long>, greater<long long> > small;
    int m;
    for(int i = 0; i < n; i++)
    {
        scanf("%d", &m);
        big.push(m);
        small.push(m);
    }
    //这里我们要凑一下，以便每次都能合并k堆且最后能合并成一堆
    while(small.size() % (k - 1) != 1) small.push(0);
    long long ans1 = 0;
    while(big.size() > 1)
    {
        long long a = big.top();
        big.pop();
        long long b = big.top();
        big.pop();
        ans1 += a + b;
        big.push(a + b);
    }
    long long ans2 = 0;
    while(small.size() > 1)
    {
        long long tmp = 0;
        for(int i = 0; i < k; i++)
        {
            tmp += small.top();
            small.pop();
        }
        ans2 += tmp;
        small.push(tmp);
    }
    printf("%lld %lld", ans1, ans2);
    return 0;
}