Leetcode 1306.跳跃游戏 III（Jump Game III）

Leetcode 1306.跳跃游戏 III

1 题目描述（Leetcode题目链接）

  这里有一个非负整数数组 arr，你最开始位于该数组的起始下标 start 处。当你位于下标 i 处时，你可以跳到 i + arr[i] 或者 i - arr[i]。

请你判断自己是否能够跳到对应元素值为 0 的 任意 下标处。

注意，不管是什么情况下，你都无法跳到数组之外。

输入：arr = [4,2,3,0,3,1,2], start = 5
输出：true
解释：
到达值为 0 的下标 3 有以下可能方案： 
下标 5 -> 下标 4 -> 下标 1 -> 下标 3 
下标 5 -> 下标 6 -> 下标 4 -> 下标 1 -> 下标 3 

输入：arr = [3,0,2,1,2], start = 2
输出：false
解释：无法到达值为 0 的下标 1 处。 

提示：

• $1\le arr.length\le 5\ast 10^4$
• $0\le arr[i]<arr.length$
• $0\le start<arr.length$

2 题解

  深度优先或广度优先都可以解决。
广度优先（BFS）：

class Solution:
    def canReach(self, arr: List[int], start: int) -> bool:
        clevel = [start]
        asked = {start}
        length = len(arr)
        while clevel:
            next_level = []
            for index in clevel:
                if arr[index] == 0:
                    return True
                for j in index - arr[index], index + arr[index]:
                    if j not in asked and length > j >= 0:
                        asked.add(j)
                        next_level.append(j)
            clevel = next_level
        return False

深度优先（DFS）:

class Solution:
    def canReach(self, arr: List[int], start: int) -> bool:
        asked = set()
        length = len(arr)
        def dfs(index):
            if index < 0 or index >= length:
                return False
            if arr[index] == 0:
                return True
            if index not in asked:
                asked.add(index)
                a = arr[index] + index
                b = index - arr[index]
                return dfs(a) or dfs(b)
            return False
        return dfs(start)