1836：【04NOIP提高组】合并果子(三种做法)

Ia_aIweq

已于 2022-10-07 21:28:14 修改

阅读量3.2k

点赞数 6

CC 4.0 BY-SA版权

分类专栏：贪心算法树队列文章标签：算法 c++ 开发语言

于 2022-10-06 09:18:31 首次发布

本文链接：https://blog.csdn.net/weq2011/article/details/127179585

树同时被 3 个专栏收录

32 篇文章

订阅专栏

贪心算法

7 篇文章

订阅专栏

队列

5 篇文章

订阅专栏

本文介绍了解决果子合并问题的三种方法：贪心算法、堆和优先队列。通过实例代码详细展示了每种方法的具体实现过程，旨在帮助读者理解不同数据结构在解决同一问题时的应用。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

一，贪心

每次只要合并果子数量最小的两堆,再计算耗费的体力值即可。

代码:

#include <bits/stdc++.h>
using namespace std;
int ans,n,a[100001];
int main()
{
  cin>>n;
  for(int i = 0; i < n; i++) cin>>a[i];
  sort(a,a + n);
  while(n > 1)
  {
    a[1] += a[0];//合并最小的两堆
    ans += a[1];//消耗相应体力值
    for(int j = 1; j < n; j++) a[j - 1] = a[j];// 删除第一个元素
    n--;//长度--
    sort(a,a + n);//再次排序
  }
  cout<<ans;
  return 0;
}

二，堆

每输入一个数就把它push进一个小顶堆，然后将最小的两个元素相加得到新的一堆(两个弹出操作）,再消耗相应的体力值，再把新的一堆push进小顶堆中。

代码:

#include <bits/stdc++.h>
using namespace std;
int ans,n,a[100001],sie = 0;//答案  输入果子种类的个数   长度
void push(int n)
{
  a[++sie] = n;
  push_heap(a + 1,a + 1 + sie,greater<int>());//把n push进最小堆
}
int pop()
{
  pop_heap(a + 1,a + 1 + sie,greater<int>());//弹出堆顶元素
  return a[sie--];//再返回堆顶数值
}
int main()
{
  cin>>n;
  for(int i = 1; i <= n; i++)
  {
    int t;
    cin>>t;
    push(t);//建立最小堆
  }
  while(sie > 1)
  {
    int s = pop() + pop();//将最小的两个元素相加得到新的一堆
    ans += s;//消耗相应的体力值
    push(s);//再把新的一堆push进小顶堆中
  }
  cout<<ans;
  return 0;
}

三，优先队列

#include <bits/stdc++.h>
using namespace std;
int ans,n,a[100001],sie = 0;
priority_queue<int,vector<int>,greater<int>> q;
int main()
{
  cin>>n;
  for(int i = 1; i <= n; i++)
  {
    int t;
    cin>>t;
    q.push(t);
    sie++;
  }
  while(sie > 1)
  {
    int s = q.top();
    q.pop();
    s += q.top();
    q.pop();
    ans += s;
    q.push(s);
    sie--;
  }
  cout<<ans;
  return 0;
}

关于greater的用法,详见C++ std::greater用法及代码示例_诸葛灬孔暗的博客-CSDN博客_c++ greater