Leetcode1306. 跳跃游戏 III(medium,BFS,DFS)

目录

1. 问题描述

2. 解题分析

3. 代码实现


1. 问题描述

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

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

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

示例 1:

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

示例 2:

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

示例 3:

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

提示:

  • 1 <= arr.length <= 5 * 10^4
  • 0 <= arr[i] < arr.length
  • 0 <= start < arr.length

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/jump-game-iii
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

2. 解题分析

        应该就是普通的深度或广度优先搜索都可以吧。

         在每个节点处,可以向左跳或者向右跳,但是要满足不能跳出去(即越界),并且不要重复访问(重复访问没有意义)。

3. 代码实现

from typing import List
from collections import deque
class Solution:
    def canReach(self, arr: List[int], start: int) -> bool:
        
        q = deque([start])
        visited = set([start])
        while len(q) > 0:
            node = q.pop()
            if arr[node] == 0:
                return True
            left = node - arr[node]
            if left >= 0 and left not in visited:
                q.append(left)
                visited.add(left)
            
            right = node + arr[node]
            if right < len(arr) and right not in visited:
                q.append(right)
                visited.add(right)

        return False
    
if __name__ == "__main__":
    
    sln = Solution()
    
    arr = [4,2,3,0,3,1,2]
    start = 5
    print(sln.canReach(arr, start))
    
    arr = [4,2,3,0,3,1,2]
    start = 0
    print(sln.canReach(arr, start))
    
    arr = [3,0,2,1,2]
    start = 2
    print(sln.canReach(arr, start))
    
    arr = [0,1]
    start = 1
    print(sln.canReach(arr, start))

        执行用时:52 ms, 在所有 Python3 提交中击败了94.80%的用户

        内存消耗:20.3 MB, 在所有 Python3 提交中击败了45.97%的用户

        嗯,随手一写,有这个表现也可以满意了。

        回到主目录:笨牛慢耕的Leetcode每日一解题解笔记(动态更新。。。) 

猜你喜欢

转载自blog.csdn.net/chenxy_bwave/article/details/124240654
今日推荐