LeetCode-45、跳跃游戏 II-困难

LeetCode-45、跳跃游戏 II-困难

给定一个非负整数数组，你最初位于数组的第一个位置。

数组中的每个元素代表你在该位置可以跳跃的最大长度。

你的目标是使用最少的跳跃次数到达数组的最后一个位置。

 

示例：

输入: [2,3,1,1,4]
输出: 2
解释: 跳到最后一个位置的最小跳跃数是 2。
     从下标为 0 跳到下标为 1 的位置，跳 1 步，然后跳 3 步到达数组的最后一个位置。

说明：

假设你总是可以到达数组的最后一个位置。

 

代码1：超出时间限制

# 找出所有可能路径，判断最短路径
class Solution:
    def jump(self, nums: List[int]) -> int:
        res = []
        nlen = len(nums)
        
        def backtrack(ind, count):
            cur = nums[ind]
            if ind+cur >= nlen-1:
                res.append(count+1)
            else:
                while cur:
                    backtrack(ind+cur, count+1)
                    cur -= 1
            return
       
        if nlen <= 1:
            return 0
        else:
            backtrack(0, 0)
            return min(res)

 

代码2：BFS解法（剪枝）

class Solution:
    def jump(self, nums: List[int]) -> int:
        nlen = len(nums)
        if nlen == 1:
            return 0
        
        res = [(0, 0)]
        visited = [False] * nlen
        while res:                             # 或写为：‘while 1:’
            count, curind = res.pop(0)
            for i in range(nums[curind], 0, -1):
                if curind+i >= nlen-1:
                    return count+1
                if not visited[curind+i]:
                    res.append((count+1, curind+i))
                    visited[curind+i] = True

参考：python BFS解法