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

#include<bits/stdc++.h>

using namespace std;

int main()

{

int n,k;

scanf("%d%d",&n,&k);

priority_queue<int >queue1;//

priority_queue<int ,vector<int >, greater<int > >queue2;

for(int i=0;i<=n-1;i++)

{

int x;

scanf("%d",&x);

queue1.push(x);

queue2.push(x);

}

long long sum1 = 0;

while(queue2.size()%(k-1)!=1) queue2.push(0);

while(queue2.size()>1)

{

long long sumsum = 0;

for(int j=1;j<=k;j++)

{

int aa=queue2.top();queue2.pop();sumsum+=aa;

}

sum1+=sumsum;

queue2.push(sumsum);

}

long long sum2 = 0;

while(queue1.size()>1)

{

int aa=queue1.top();

queue1.pop();

int bb=queue1.top();

queue1.pop();

sum2+=aa+bb;

queue1.push(aa+bb);

}

printf("%ld %ld\n",sum2,sum1);

return 0;

}