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
最大的花费,就要找最大的数重叠次数最多,所以与K无关,只需要从最大的开始每次都选最大的两堆来相加就可以得出最大的重叠次数。
最小的花费思路和最大的相同,要重迭次数最小,就要挑选K最大,每次从最小的K堆,开始合并。这里要做一步处理是如果最后不够K堆合并的时候要在整个队列的前面填0(填零的意思就是一开始不按照K堆合并,而是按照从0-K的一个数合并,这样后面的数重叠次数才是最小的,如果不补零,后面大的数就会多重叠)。
#include <iostream>
#include <algorithm>
#include <queue>
using namespace std;
int main()
{
priority_queue <int> q1; //构建最大优先队列 百度一下优先队列很简单
priority_queue <int, vector<int>, greater<int> > q2;//构建最小优先队列
int n, i, a, k;
cin>>n>>k;
for(i = 0; i < n; i++)
{
cin >> a;
q1.push(a);
q2.push(a);
}
long long minn = 0, maxx = 0;
long long sum1 = 0, sum2 = 0;
while(q2.size() % (k - 1) != 1)
{
q2.push(0);
}
while(q2.size() > 1)
{
sum2 = 0;
for(i = 0; i < k; i++)
{
sum2 += q2.top();
q2.pop();
}
minn += sum2;
q2.push(sum2);
}
while(q1.size() > 1)
{
sum1 = 0;
for(i = 0; i < 2; i++)
{
sum1 += q1.top();
q1.pop();
}
maxx += sum1;
q1.push(sum1);
}
cout<<maxx<<' '<<minn<<endl;
return 0;
}