-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy path055_canJump.py
45 lines (38 loc) · 1.36 KB
/
055_canJump.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
'''
给定一个非负整数数组,你最初位于数组的第一个位置。
数组中的每个元素代表你在该位置可以跳跃的最大长度。
判断你是否能够到达最后一个位置。
示例 1:
输入: [2,3,1,1,4]
输出: true
解释: 我们可以先跳 1 步,从位置 0 到达 位置 1, 然后再从位置 1 跳 3 步到达最后一个位置。
示例 2:
输入: [3,2,1,0,4]
输出: false
解释: 无论怎样,你总会到达索引为 3 的位置。但该位置的最大跳跃长度是 0 , 所以你永远不可能到达最后一个位置。
'''
from typing import List
class Solution:
def canJump(self, nums: List[int]) -> bool:
nums[-1] = len(nums)-1
for i in range(len(nums)-1,-1,-1):
if nums[i] == 0:
nums[i] = i
elif i + nums[i] >= len(nums) - 1:
nums[i] = len(nums)-1
else:
index = nums[i]
nums[i] = max(nums[i+1:i+index+1])
if nums[0] == len(nums) - 1:
return True
else:
return False
# class Solution:
# def canJump(self, nums: List[int]) -> bool:
# index = len(nums) - 1
# for i in range(len(nums)-2,-1,-1):
# if nums[i] + i >= index:
# index = i
# return index == 0
a = Solution()
print(a.canJump([2,3,1,0,2]))