给定一个非负整数数组 nums ,你最初位于数组的 第一个下标 。
数组中的每个元素代表你在该位置可以跳跃的最大长度。
判断你是否能够到达最后一个下标。
示例 1:
输入:nums = [2,3,1,1,4]
输出:true
解释:可以先跳 1 步,从下标 0 到达下标 1, 然后再从下标 1 跳 3 步到达最后一个下标。
示例 2:
输入:nums = [3,2,1,0,4]
输出:false
解释:无论怎样,总会到达下标为 3 的位置。但该下标的最大跳跃长度是 0 , 所以永远不可能到达最后一个下标。
提示:
1 <= nums.length <= 3 * 104
0 <= nums[i] <= 105
来源:力扣(LeetCode)
链接:力扣https://leetcode.cn/problems/jump-game/先给出一个不用动态规划的简单想法:
我们遍历数组中的每一个元素,如果其中不包含0,那么就非True无疑了,其他情况下,第一个元素是0,但是列表中元素不止一个,必定是False,最后一种情况下,我们找到0,再把0前元素的值与下标的差来作比较即可判断。
class Solution:
def canJump(self, nums: List[int]) -> bool:
if 0 not in nums:
return True
elif nums[0]==0 and len(nums)>1:
return False
else:
k=0
diy=True
for j in range(len(nums)):
if nums[j]==0:
for i in range(k,j):
if (nums[i]>j-i) or (nums[i]==j-i and j==len(nums)-1):
diy=True
break
else:
diy=False
if diy==False:
return False
return True
用时并不理想。
又想了一个简单的动态规划的算法,但是有许多重复操作,导致最后提交还超时了。
class Solution:
def canJump(self, nums: List[int]) -> bool:
n = len(nums)
if n==1:
return True
ls = [False] * n
for i in range(n):
if nums[i] != 0:
for j in range(i, i + nums[i] + 1):
if j <= n - 1:
ls[j] = True
else:
if n> i+1 and ls[i+1] == False:
break
return not False in ls
最终动态规划改良后终于通过了。
class Solution:
def canJump(self, nums: List[int]) -> bool:
n=len(nums)
mi=0
for i in range(n):
mi=max(mi,i+nums[i])
if mi>=n-1:
return True
if mi<i+1:
return False