leet55. 跳跃游戏

题目:

给定一个非负整数数组,您最初位于数组的第一个索引处。

数组中的每个元素表示您在该位置的最大跳跃长度。

确定是否能够到达最后一个索引。

示例:
A = [2,3,1,1,4],返回 true

A = [3,2,1,0,4],返回 false

分析:

  1. 该问题是动态规划问题;
  2. 如果除尾部元素外的所有元素均为非零元素,则一定可以到达最后一个索引
  3. 当除尾部元素外,其余元素存在0元素,则要跳过该0元素,在0之前需要有一个元素,足够使得从大元素位置可以跳过该0元素位置
  4. 为实现以上目的,建立一个栈,用于存储0元素及其出现后的元素,当入栈元素值大于栈长度减1时,表明由该位置可以跳过栈底的0元素,则将栈清空
  5. 逆向遍历列表后,判断如果为空栈则可以到达最后一个索引

代码:

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

思考:


猜你喜欢

转载自blog.csdn.net/yzp1011/article/details/79764831
今日推荐