堆:合并果子

这是一个关于计算合并果子最小体力耗费的算法问题。由于数据范围较大,不能直接使用优先队列,而是通过两个队列模拟合并过程。程序首先使用桶排序将果子数量放入队列,然后进行n-1次合并,每次合并取两个队列中最小的堆,更新答案并放入新的队列。最后输出最小的体力耗费值。代码中展示了如何实现这一过程。
摘要由CSDN通过智能技术生成

题目链接: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;
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值