文章目录
一、题目
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、原题链接
二、解题报告
1、思路分析
(
1
)
(1)
(1) 如果我反复横跳,先去左边,再去右边,再去左边,再去右边。如图所示:
(
2
)
(2)
(2) 我们发现一个很重要的点,就是一个位置,只要被至少一个线段射线,就可以拿到水果了,多余的覆盖是没有意义的。
(
3
)
(3)
(3) 所以,上面这个图,它一定不会比下面这种情况更优:
(
4
)
(4)
(4) 于是,我们想到一个想法,就是每次我可以选择左边做到某个点,然后立即回头走完剩下的路。
(
5
)
(5)
(5) 这就像我们的人生,不要光顾着一直往前冲,偶尔也要停下来,回头看看沿途的风景,沿途的风景或许比终点更加美好。
(
6
)
(6)
(6) 所以,我们枚举往左能够到达的所有点,那么剩下往右的那段路程,是不是可以通过二分枚举快速的求出来。
(
7
)
(7)
(7) 于是,算法就变成了 二分枚举 + 前缀和。
2、算法详解
(
1
)
(1)
(1) 预处理水果总数前缀和;
(
2
)
(2)
(2) 处理前缀和的目的是为了利用差分计算部分和(区间和),区间和就会涉及到减法,并且下标有可能为 -1,所以我们把下标从 1 开始算,或者抽象出一个函数来;
3、时间复杂度
最坏时间复杂度 O ( n l o g n ) O(nlogn) 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
)
(1)
(1) 如果当前枚举的位置已经大于起始位置,则不可能是先向左的,直接跳出循环;
(
2
)
(2)
(2) 否则,先去掉向左走的步数,如果步数已经小于 0,则不可能向右走了。
(
3
)
(3)
(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讲》🌌
几张动图学会一种数据结构 🌳《画解数据结构》🌳
竞赛选手金典图文教程 💜《夜深人静写算法》💜