LeetCode 2106. 摘水果

一、题目

1、题目描述

  在一个无限的 x坐标轴上，有许多水果分布在其中某些位置。给你一个二维整数数组 fruits，其中 fruits[i] = [positioni, amounti]表示共有 amounti个水果放置在 positioni上。fruits已经按 positioni升序排列 ，每个 positioni互不相同 。
  另给你两个整数 startPos和 k。最初，你位于 startPos。从任何位置，你可以选择 向左 或者 向右 走。在 x轴上每移动 一个单位 ，就记作 一步 。你总共可以走最多 k步。你每达到一个位置，都会摘掉全部的水果，水果也将从该位置消失（不会再生）。
  返回你可以摘到水果的 最大总数 。
  样例输入： fruits = [[2,8],[6,3],[8,6]], startPos = 5, k = 4
  样例输出： 9

2、基础框架

• C语言 给出的基础框架代码如下：

int maxTotalFruits(int** fruits, int fruitsSize, int* fruitsColSize, int startPos, int k){

}

3、原题链接

LeetCode 2106. 摘水果

二、解题报告

1、思路分析

  $(1)$ 如果我反复横跳，先去左边，再去右边，再去左边，再去右边。如图所示：

  $(2)$ 我们发现一个很重要的点，就是一个位置，只要被至少一个线段射线，就可以拿到水果了，多余的覆盖是没有意义的。
  $(3)$ 所以，上面这个图，它一定不会比下面这种情况更优：

  $(4)$ 于是，我们想到一个想法，就是每次我可以选择左边做到某个点，然后立即回头走完剩下的路。
  $(5)$ 这就像我们的人生，不要光顾着一直往前冲，偶尔也要停下来，回头看看沿途的风景，沿途的风景或许比终点更加美好。
  $(6)$ 所以，我们枚举往左能够到达的所有点，那么剩下往右的那段路程，是不是可以通过二分枚举快速的求出来。
  $(7)$ 于是，算法就变成了 二分枚举 + 前缀和。

2、算法详解

  $(1)$ 预处理水果总数前缀和；
  $(2)$ 处理前缀和的目的是为了利用差分计算部分和（区间和），区间和就会涉及到减法，并且下标有可能为 -1，所以我们把下标从 1 开始算，或者抽象出一个函数来；

3、时间复杂度

   最坏时间复杂度 $O(nlogn)$ 。

4、代码详解

1）差分

  为了避免数组下标越界，所以差分的时候，需要考虑左边界是 0 的情况，直接返回 sum[r]；否则返回sum[r] - sum[l-1]。

int sum[200010];

int getSum(int l, int r) {
    if(l == 0) {
        return sum[r];
    }
    return sum[r] - sum[l-1];
}

2）前缀和的预处理

void preCalc(int** fruits, int n) {
    for(int i = 0, cnt = 0; i < n; ++i) {
        cnt = fruits[i][1];
        sum[i] = cnt;
        if(i) {
            sum[i] += sum[i-1];
        }
    }
}

3）先向左，再向右

  $(1)$ 如果当前枚举的位置已经大于起始位置，则不可能是先向左的，直接跳出循环；
  $(2)$ 否则，先去掉向左走的步数，如果步数已经小于 0，则不可能向右走了。
  $(3)$ 否则，就是利用二分查找，找到向右走能够走到的最远的点。

    for(i = 0; i < fruitsSize; ++i) {
        pos = fruits[i][0];
        if(pos > startPos) {
            break;                  // (1)
        }
        step = k;
        // 先走 startPos - pos 的步数
        step -= (startPos - pos);
        if(step < 0) {
            continue;               // (2)
        }
        // 单纯往左走
        ret = max(ret, getSum(i, lbound));
        l = i;                      // (3)
        r = fruitsSize - 1;
        while(l <= r) {
            m = (l + r) >> 1;
            if( fruits[m][0] - pos <= step ) {
                ans = m;
                l = m + 1;
            } else {
                r = m - 1;
            }
        }
        ret = max(ret, getSum(i, ans));   // (4)
    }

4）先向右，再向左

  同理可得。

    for(i = fruitsSize - 1; i >= 0; --i) {
        pos = fruits[i][0];
        if(pos < startPos) {
            break;
        }
        step = k;
        // 先走 pos - startPos 的步数
        step -= (pos - startPos);
        if(step < 0) {
            continue;
        }
        // 单纯往右走的情况
        ret = max(ret, getSum(rbound, i) );
        l = 0;
        r = i;
        while(l <= r) {
            m = (l + r) >> 1;
            if( pos - fruits[m][0] <= step ) {
                ans = m;
                r = m - 1;
            } else {
                l = m + 1;
            }
        }
        ret = max(ret, getSum(ans, i));
    }

---

三、本题小知识

  看到有序，想到二分。

---

四、加群须知

  相信看我文章的大多数都是「 大学生 」，能上大学的都是「 精英 」，那么我们自然要「 精益求精 」，如果你还是「 大一 」，那么太好了，你拥有大把时间，当然你可以选择「 刷剧 」，然而，「 学好算法 」，三年后的你自然「 不能同日而语 」。
  那么这里，我整理了「 几十个基础算法 」 的分类，点击开启：

🌌《算法入门指引》🌌
  如果链接被屏蔽，或者有权限问题，可以私聊作者解决。

  大致题集一览：

  为了让这件事情变得有趣，以及「 照顾初学者 」，目前题目只开放最简单的算法 「 枚举系列 」 （包括：线性枚举、双指针、前缀和、二分枚举、三分枚举），当有 一半成员刷完 「 枚举系列 」 的所有题以后，会开放下个章节，等这套题全部刷完，你还在群里，那么你就会成为「 夜深人静写算法 」专家团 的一员。
  不要小看这个专家团，三年之后，你将会是别人 望尘莫及 的存在。如果要加入，可以联系我，考虑到大家都是学生， 没有「 主要经济来源 」，在你成为神的路上，「 不会索取任何 」。
  🔥联系作者，或者扫作者主页二维码加群，加入刷题行列吧🔥

---

🔥让天下没有难学的算法🔥
C语言免费动漫教程，和我一起打卡！ 🌞《光天化日学C语言》🌞
让你养成九天持续刷题的习惯 🔥《九日集训》🔥
入门级C语言真题汇总 🧡《C语言入门100例》🧡
组团学习，抱团生长 🌌《算法零基础100讲》🌌
几张动图学会一种数据结构 🌳《画解数据结构》🌳
竞赛选手金典图文教程 💜《夜深人静写算法》💜