跳跃游戏(python)

链接: https://leetcode.cn/problems/jump-game
题目描述:
给定一个非负整数数组 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 ∗ 1 0 4 3 * 10^4 3104
  • 0 <= nums[i] <= 1 0 5 10^5 105

思路

只要最后能到达的位置的跳跃长度不为0,就能一直跳下去。所以只要判断max_pos的nums[i]是否为0就可以了

代码

class Solution:
    def canJump(self, nums: List[int]) -> bool:
        max_pos=0
        if len(nums)==1:
            return True
        for i in range(len(nums)-1):
            max_pos=max(max_pos,nums[i]+i)
            if max_pos>=len(nums)-1:
                return True
            if nums[i]==0 and max_pos==i:
                return False

把True写在里边可以减少一点运行时间

class Solution:
    def canJump(self, nums: List[int]) -> bool:
        max_pos=0
        for i in range(len(nums)-1):
            max_pos=max(max_pos,nums[i]+i)
            if nums[i]==0 and max_pos==i:
                return False
        return True

你可能感兴趣的:(leetcode,算法)