题目:
给定一个非负整数数组,您最初位于数组的第一个索引处。
数组中的每个元素表示您在该位置的最大跳跃长度。
确定是否能够到达最后一个索引。
示例:
A = [2,3,1,1,4]
,返回 true
。
A = [3,2,1,0,4]
,返回 false
。
分析:
- 该问题是动态规划问题;
- 如果除尾部元素外的所有元素均为非零元素,则一定可以到达最后一个索引
- 当除尾部元素外,其余元素存在0元素,则要跳过该0元素,在0之前需要有一个元素,足够使得从大元素位置可以跳过该0元素位置
- 为实现以上目的,建立一个栈,用于存储0元素及其出现后的元素,当入栈元素值大于栈长度减1时,表明由该位置可以跳过栈底的0元素,则将栈清空
- 逆向遍历列表后,判断如果为空栈则可以到达最后一个索引
代码:
class Solution(object):
def canJump(self, nums):
"""
:type nums: List[int]
:rtype: bool
"""
stack = []
zeroStack = []
for i,x in enumerate(nums[::-1]):
# print stack
if not x:
if i > 0:
stack.append(x)
zeroStack.append(x)
else:
if stack:
stack.append(x)
if stack[-1] > len(stack) - 1:
stack = []
if stack:
return False
return True
思考: