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;
}