给定一个非负整数数组,你最初位于数组的第一个位置。
数组中的每个元素代表你在该位置可以跳跃的最大长度。
判断你是否能够到达最后一个位置。
示例 1:
输入: [2,3,1,1,4]
输出: true
解释: 从位置 0 到 1 跳 1 步, 然后跳 3 步到达最后一个位置。
示例 2:
输入: [3,2,1,0,4]
输出: false
解释: 无论怎样,你总会到达索引为 3 的位置。但该位置的最大跳跃长度是 0 , 所以你永远不可能到达最后一个位置。
解题思路
这个问题可以通过递归很容易解决来处理,我们想要知道能否到达index
,那么只需要知道index
之前的元素是不是有nums[i]+i >= index for i in range(index)
(也就是index
之前是不是有位置可以到达index
)。例如
3 2 1 0 4
|<- i ->| index
如果有的话,那么我们只需要判断是否有元素能到i
(递归下去)即可,否则我们返回False
。我们可以定义函数f(x)
表示位置x
是不是可以到达,那么
- f ( x ) = n u m s [ i ] + i > = i n d e x & & f ( i ) f(x)=nums[i]+i>=index\ \&\& \ f(i) f(x)=nums[i]+i>=index && f(i)
class Solution:
def canJump(self, nums):
"""
:type nums: List[int]
:rtype: bool
"""
nums_len = len(nums)
return self._canJump(nums, nums_len - 1)
def _canJump(self, nums, index):
if index == 0:
return True
for i in range(index):
if nums[i] + i >= index and self._canJump(nums, i):
return True
return False
我们也很容易通过记忆化搜索的方法对这个问题优化
class Solution:
def canJump(self, nums):
"""
:type nums: List[int]
:rtype: bool
"""
nums_len = len(nums)
mem = [None]*nums_len
mem[0] = True
return self._canJump(nums, nums_len - 1, mem)
def _canJump(self, nums, index, mem):
if mem[index] != None:
return mem[index]
for i in range(index):
if nums[i] + i >= index and self._canJump(nums, i, mem):
mem[i] = True
return True
mem[i] = False
return False
但是这种做法在提交的时候超时了,我们只能想其他的办法。
对于这个问题最好的解法是贪心算法。对于index
的位置,我们将index
之前的元素值和其索引值相加
3 2 1 0 4
3 3 3 3 index
然后我们计算之前和的最大值m
,如果index>m
,那么自然不能到达index
了。对于所有[0:len(nums))
都满足上述条件的话,那么自然就成立了。
总体的思路就是,我们希望前面的步子迈大一点,这样可以给后面留有更多的余地。
class Solution:
def canJump(self, nums):
"""
:type nums: List[int]
:rtype: bool
"""
m = 0
for i, num in enumerate(nums):
if i > m:
return False
m = max(m, i + num)
return True
我将该问题的其他语言版本添加到了我的GitHub Leetcode
如有问题,希望大家指出!!!