题目背景
本题除【数据范围与约定】外与 P1090 完 全 一 致。
题目描述
在一个果园里,多多已经将所有的果子打了下来,而且按果子的不同种类分成了不同的堆。多多决定把所有的果子合成一堆。
每一次合并,多多可以把两堆果子合并到一起,消耗的体力等于两堆果子的重量之和。可以看出,所有的果子经过 (n - 1)(n−1) 次合并之后, 就只剩下一堆了。多多在合并果子时总共消耗的体力等于每次合并所耗体力之和。
因为还要花大力气把这些果子搬回家,所以多多在合并果子时要尽可能地节省体力。假定每个果子重量都为 11,并且已知果子的种类数和每种果子的数目,你的任务是设计出合并的次序方案,使多多耗费的体力最少,并输出这个最小的体力耗费值。
例如有 33 堆果子,数目依次为 1,~2,~91, 2, 9。可以先将 11、22 堆合并,新堆数目为 33,耗费体力为 33。接着,将新堆与原先的第三堆合并,又得到新的堆,数目为 1212,耗费体力为 1212。所以多多总共耗费体力为 3+12=153+12=15。可以证明 1515 为最小的体力耗费值。
输入格式
输入的第一行是一个整数 nn,代表果子的堆数。
输入的第二行有 nn 个用空格隔开的整数,第 ii 个整数代表第 ii 堆果子的个数 a_iai。
输出格式
输出一行一个整数,表示最小耗费的体力值。
输入输出样例
输入 #1复制
3 1 2 9
输出 #1复制
15
说明/提示
【数据规模与约定】
本题采用多测试点捆绑测试,共有四个子任务。
- Subtask 1(10 points):1 \leq n \leq 81≤n≤8。
- Subtask 2(20 points):1 \leq n \leq 10^31≤n≤103。
- Subtask 3(30 points):1 \leq n \leq 10^51≤n≤105。
- Subtask 4(40 points):1 \leq n \leq 10^71≤n≤107。
对于全部的测试点,保证 1 \leq a_i \leq 10^51≤ai≤105。
【提示】
- 请注意常数因子对程序效率造成的影响。
- 请使用类型合适的变量来存储本题的结果。
- 本题输入规模较大,请注意数据读入对程序效率造成的影响。
上代码:
#include <cstdio>
#include <queue>
#define int long long
using namespace std;
queue <int> q1;
queue <int> q2;
int to[100005];
void read(int &x){
int f=1;x=0;char s=getchar();
while(s<'0'||s>'9'){if(s=='-')f=-1;s=getchar();}
while(s>='0'&&s<='9'){x=x*10+s-'0';s=getchar();}
x*=f;
}
signed main() {
int n;
read(n);
for (int i = 1; i <= n; ++i) {
int a;
read(a);
to[a] ++;
}
for (int i = 1; i <= 100000; ++i) {
while(to[i]) {
to[i] --;
q1.push(i);
}
}
int ans = 0;
for (int i = 1; i < n; ++i) {
int x , y;
if((q1.front() < q2.front() && !q1.empty()) || q2.empty()) {
x = q1.front();
q1.pop();
}
else {
x = q2.front();
q2.pop();
}
if((q1.front() < q2.front() && !q1.empty()) || q2.empty()) {
y = q1.front();
q1.pop();
}
else {
y = q2.front();
q2.pop();
}
ans += x + y;
q2.push(x + y);
}
printf("%lld" , ans);
return 0;
}