算法作业 多元Huffman编码问题（贪心）（C++优先队列）

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

 

我一看这题，我心想直接两个优先队列把数扔进去贪心算不就完了吗，结果WA了，然后百度才发现有的样例过不了，需要前面加0，借鉴的博客是这个：https://blog.csdn.net/galesaur_wcy/article/details/84393051

 

我的代码如下：

#include<queue>
#include<iostream>
#include<algorithm>
using namespace std;

#define ll long long//可能会爆int
ll a[100005];
int main(){
    int n,k;
    cin>>n>>k;
    for(int i=0;i<n;i++){
        cin>>a[i];
    }

    priority_queue<ll,vector<ll>,greater<ll> >p;//优先队列小顶堆
    priority_queue<ll,vector<ll>,less<ll> >q;//优先队列大顶堆
    for(int i=0;i<n;i++){
        p.push(a[i]);
        q.push(a[i]);
    }
    int m = n;
    while(m>k){
        m=m-k+1;//玄学关系
    }
    for(int i=0;i<k-m;i++){//需要加的0的个数为k-m
        p.push(0);
    }
    ll minsum = 0;//最小花费
    while(p.size()>1){
        ll sum = 0;
        for(int j=0;j<k;j++){
            if(!p.empty()){
                sum+=p.top();
                p.pop();
            }
        }
        p.push(sum);
        minsum+=sum;
    }

    ll maxsum = 0;//最大花费
    while(q.size()>1){
        ll sum = 0;
        for(int j=0;j<2;j++){
            if(!q.empty()){
                sum+=q.top();
                q.pop();
            }
        }
        q.push(sum);
        maxsum+=sum;
    }
    cout<<maxsum<<" "<<minsum<<endl;
    return 0;
}