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

#include <iostream>
#include <cstdio>
#include <queue>
#include <vector>
using namespace std;
priority_queue<int >queue1; ///默认降序
priority_queue<int,vector<int >,greater<int> >queue2;
void greedy(int n,int k)
{
    long long sum1 = 0,sum2 = 0,heap = 0;
    while(queue2.size()%(k-1)!=1)
    {
        queue2.push(0);
    }
    ///求最大
    while(queue1.size()>1)
    {
        int a =queue1.top();
        queue1.pop();
        int b = queue1.top();
        queue1.pop();
        heap = a+b;
        queue1.push(heap);
        sum1+=heap;
    }
///求最小
    heap = 0;
    while(queue2.size()>1)
    {
        for(int i = 1; i <= k; i++)
        {
            heap+=queue2.top();
            queue2.pop();
        }
        sum2+=heap;
        queue2.push(heap);
        heap = 0;   ///记得清零哦!
    }
        cout<<sum1<<" "<<sum2<<endl;
}
int main()
{
    int n,k,x;
    scanf("%d %d",&n,&k);
    for(int i = 0; i < n; i++)
    {
        cin>>x;
        queue1.push(x);
        queue2.push(x);
    }
    greedy(n,k);
    return 0;
}