题目链接:https://www.luogu.com.cn/problem/P6033
题目:
题目背景
本题除【数据范围与约定】外与 P1090 完 全 一 致。
题目描述
在一个果园里,多多已经将所有的果子打了下来,而且按果子的不同种类分成了不同的堆。多多决定把所有的果子合成一堆。
每一次合并,多多可以把两堆果子合并到一起,消耗的体力等于两堆果子的重量之和。可以看出,所有的果子经过 (n - 1)次合并之后, 就只剩下一堆了。多多在合并果子时总共消耗的体力等于每次合并所耗体力之和。
因为还要花大力气把这些果子搬回家,所以多多在合并果子时要尽可能地节省体力。假定每个果子重量都为 1,并且已知果子的种类数和每种果子的数目,你的任务是设计出合并的次序方案,使多多耗费的体力最少,并输出这个最小的体力耗费值。
例如有 3 堆果子,数目依次为 1,~2,~91, 2, 9。可以先将 1、2 堆合并,新堆数目为 3,耗费体力为 3。接着,将新堆与原先的第三堆合并,又得到新的堆,数目为 12,耗费体力为 12。所以多多总共耗费体力为 3+12=15。可以证明 15 为最小的体力耗费值。
输入格式
输入的第一行是一个整数 n,代表果子的堆数。
输入的第二行有 n 个用空格隔开的整数,第 i 个整数代表第 i 堆果子的个数 ai。输出格式
输出一行一个整数,表示最小耗费的体力值。
输入输出样例
输入 #1复制
3 1 2 9输出 #1复制
15说明/提示
【数据规模与约定】
本题采用多测试点捆绑测试,共有四个子任务。
- 1≤n≤107。
对于全部的测试点,保证 1≤ai≤105。
【提示】
- 请注意常数因子对程序效率造成的影响。
- 请使用类型合适的变量来存储本题的结果。
- 本题输入规模较大,请注意数据读入对程序效率造成的影响。
分析:
此题由于数据范围比较大的原因,所以直接使用优先队列的方式的话会超时。必须得将其转为0(N)的复杂度之内。所以我们将其转为使用两个队列的方式。
其中,此题有一个很重要的地方,就是a1 <= 1e5 , 所以可以使用桶排序的方式,将其保存在其中一个队列q1中。
而每次两个堆合并的新堆就放到队列q2中。而每次取最小的值,只需要从q1和q2的堆顶处,比较两个值哪个小就取哪一个。
取出两个后,合并,然后放到队列q2中。
代码实现:
# include <iostream>
# include <queue>
using namespace std;
const int N = 1e5 + 10;
int total[N];
queue<long long> q1;
queue<long long> q2;
int n ;
inline int read()
{
int x=0;char ch=getchar();
while(ch<'0'||ch>'9')ch=getchar();
while(ch>='0'&&ch<='9')x=(x<<1)+(x<<3)+(ch^48),ch=getchar();
return x;
}
int main()
{
//scanf("%d",&n);
n = read();
for(int i = 1 ; i <= n ; i++)
{
int temp;
//scanf("%d",&temp);
temp = read();
total[temp]++;
}
for(int i = 1 ; i <= 1e5 ; i++)
{
while(total[i] > 0)
{
q1.push(i);
total[i]--;
}
}
long long ans = 0;
for(int i = 1 ; i <= n - 1; i++) // 进行n - 1次合并,变为1堆
{
long long t1,t2;
if(q1.empty())
{
t1 = q2.front();
q2.pop();
}
else if(q2.empty())
{
t1 = q1.front();
q1.pop();
}
else
{
if(q1.front() <= q2.front())
{
t1 = q1.front();
q1.pop();
}
else
{
t1 = q2.front();
q2.pop();
}
}
if(q1.empty())
{
t2 = q2.front();
q2.pop();
}
else if(q2.empty())
{
t2 = q1.front();
q1.pop();
}
else
{
if(q1.front() <= q2.front())
{
t2 = q1.front();
q1.pop();
}
else
{
t2 = q2.front();
q2.pop();
}
}
ans += t1 + t2;
q2.push(t1 + t2);
}
printf("%lld\n",ans);
return 0;
}