机器人跳跃问题c++

凌窍

已于 2024-02-29 13:58:54 修改

阅读量245

点赞数 4

分类专栏：蓝桥杯辅导课之前缀和与二分文章标签： c++ 算法

于 2024-02-28 19:38:46 首次发布

本文链接：https://blog.csdn.net/m0_68416190/article/details/136353354

版权

蓝桥杯辅导课之前缀和与二分专栏收录该内容

5 篇文章 0 订阅

订阅专栏

题目

输入样例1：

5
3 4 3 2 4

输出样例1：

输入样例2：

3
4 4 4

输出样例2：

输入样例3：

3
1 6 4

输出样例3：

思路

观察数据范围，时间复杂度要控制在O(nlogn)之间，观察到样例输出均不超过最高建筑物高度，结合题目所说的“如果H(k+1)<= E, 它将得到E一H(k+1)的能量值”，可知答案E如果超过最高建筑物高度maxH，那么能量值会一直增加且为正数，但不满足题目要求的“最小初始能量”，因此答案E不可能超过最高建筑物高度maxH。

知道了E的范围，可以考虑从前往后枚举每个能量值，然后判断E是否满足“到达第N个建筑，在这个过程中能量值不能为负数个单位。”的要求，时间复杂度最坏为O(10^5 * 10^5)，超时。而由于E的取值范围可以看做是有序的，并且可以证明：当初始能量值E0大于等于某个值时，过程中能量值都不为负数。证明如下：
假设E0满足题目要求，即E1、E2、E3…Ei均大于等于0；设 ${E0}'$ >= E0,那么 ${E1}' = {E0}' - h1 \geqslant 2E0 - h1 = E1$ …… ${Ei}' \geq 0$

因此本题可以进行二分，时间复杂度最坏为O(10^5 * log10^5)，约为O(10^5 * 15)。

需要注意的是，如果初始能量值E很高，而Hi很小，能量值会一直增加，到后面甚至能溢出long long 的范围（能量值计算公式为2*…(2*(2*E -H1) - H2) …-Hi < 2^i * E），因此，当能量值大于等于maxH，那后面总的能量值肯定不会再减少，就可以停下来了。

代码

#include<bits/stdc++.h>
using namespace std;

const int N = 1e5 + 10;
int a[N], maxH;
int n;
bool check(int energy)
{
  for (int i = 1; i <= n; i ++)
  {
    energy += (energy - a[i]);
    if (energy >= maxH) return true;
    if (energy < 0) return false;
  }
  return true;
}

int main()
{
  scanf("%d",&n);
  int l = 1, r = 0;
  for (int i = 1; i <= n; i ++)
  {
    scanf("%d", &a[i]);
    r = max(r, a[i]);
  }
  maxH = r;
  
  while (l < r)
  {
    int mid = (l + r) >> 1;
    if (check(mid)) r = mid;
    else l = mid + 1;
  }
  printf("%d", l);
}