【题目描述】
As is kown, yjw 是伟大的 410 的物理课代表。但是作为物理课代表却常常要面临分卷子的任务,任务来了:
Yjw 需要N堆特定张数的卷子。摆在他面前的是一整摞卷子,这摞卷子的总张数恰好为所需要的卷子数的和。你需要把这摞卷子分成上述所需的N份。分卷子是需要耗费力气的,我们规定分X张卷子,耗费X个单位的力气。Yjw 想让他分卷子所耗费的力气最小,请你来帮帮他吧。
【输入格式】
第一行一个整数N表示 yjw 所需卷子的堆数
接下来的N行,每行一个整数,表示所需的每堆卷子的张数。
【输出格式】
一个整数,表示对卷子分了N-1次后,yjw 耗费的最小力气。
【输入样例】
3
8
5
8
【输出样例】
34
【数据范围】
N≤20000
样例解释:
起初卷子的张数为21,第一次分卷子后,耗费的力气为21,把卷子分为13和8的两堆,然后花费13个单位的力气将张数为13的那堆卷子分成张数为8和5的两堆,这样的耗费的总力气为34。
作为喜欢暴力出奇迹 的我,首先用暴力排序做了一下,结果可想而知:
其实我看见网上很多题解竟也是用时间超限方法写的 (逃
在时间超限后,我思考了一下:其实本质上就是用排序解决这道题,那么,如果用快排(sort)还是时间超限的话,其他的排序估计也会废掉。那么,就不用这些排序了。我们可以在读入数据时用堆存储,然后用堆排序(priority_queue)把此题解决。
代码:
#include <iostream>
#include <cstdio>
#include <queue>
using namespace std;
priority_queue <long long, vector<long long>, greater<long long> > q;
int main()
{
int n;
scanf ("%d", &n);
long long guo;
for (int i = 1; i <= n; i++)
{
scanf ("%lld", &guo);
q.push (guo);
}
long long heap1, heap2;
long long strength = 0;
for (int i = 1; i <= n - 1; i++)
{
heap1 = q.top();
q.pop();
heap2 = q.top();
q.pop();
strength = strength + heap1 + heap2;
q.push (heap1 + heap2);
}
printf ("%lld", strength);
return 0;
}