给定一个非负整数数组,你最初位于数组的第一个位置。
数组中的每个元素代表你在该位置可以跳跃的最大长度。
你的目标是使用最少的跳跃次数到达数组的最后一个位置。
示例:
输入: [2,3,1,1,4]
输出: 2
解释: 跳到最后一个位置的最小跳跃数是 2。
从下标为 0 跳到下标为 1 的位置,跳 1 步,然后跳 3 步到达数组的最后一个位置。
说明:
假设你总是可以到达数组的最后一个位置。
解题思路
这个问题是之前Leetcode 55:跳跃游戏(最详细的解法!!!)的提升。我们先看看这个问题能不能通过动态规划解决,只是对之前的问题稍加修改即可。
我们可以定义函数 f ( i ) f(i) f(i)表示到达位置 i i i所需的最少步数,那么我们只需要遍历前面的所有位置找到最少步数即可。
- f ( i ) = m i n ( f ( i ) , f ( j ) + 1 ) f(i)=min(f(i),f(j)+1) f(i)=min(f(i),f(j)+1)
其中 j < i j<i j<i并且 n u m s [ j ] + j > = i nums[j]+j>=i nums[j]+j>=i。
class Solution:
def jump(self, nums: List[int]) -> int:
nums_len = len(nums)
mem = [float('inf')]*nums_len
mem[0] = 0
for i in range(1,nums_len):
for j in range(i):
if nums[j] + j >= i:
mem[i] = min(mem[j]+1, mem[i])
break
return mem[-1]
但是这样写法超时了,需要更快的解法。这个问题是否可以通过贪心解决呢?
我们将nums
按照从大到小排序,也就是前面步子尽量迈大一点,后面就有更多的空间了啊?显然这种做法是不可行的,因为我们每次求解的是局部最优解,而对于全局来说就不一定是最优解了。
我们可以换一个思路,我们先思考跳一步的话最远可以跳多远,接着思考跳两步的话最远可以跳多远,以此类推,直到最远的距离大于等于nums.size()-1
,那么此时的步数自然就是最少的步数。
通过pre
记录step-1
步最多可以跳多远,cur
记录step
步最多可以跳多远。接着就是思考边界问题,对于每个index
来说:
pre < index <= cur
:那么需要再跳一步。cur >= nums.size()-1
:此时必然可以跳到终点,返回步数即可。
class Solution:
def jump(self, nums: List[int]) -> int:
n = len(nums)
pre, cur, step = -1, 0, 0
for i in range(n):
if cur >= n - 1:
return step
elif pre < i <= cur:
step, pre = step + 1, cur
cur = max(cur, i + nums[i])
return step
非常简洁,非常酷!!!
我将该问题的其他语言版本添加到了我的GitHub Leetcode
如有问题,希望大家指出!!!