Leetcode.1306 跳跃游戏 III

题目链接

这里是引用

Leetcode.1306 跳跃游戏 III Rating ： 1397

题目描述

这里有一个非负整数数组 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\ast 10^4$
• $0<=arr[i]<arr.length$
• $0<=start<arr.length$

解法：bfs

直接从起点 $start$ 开始 bfs，能否跳到一个值为 0 的点。如果可以，就返回 true；否则就返回 false。

时间复杂度： $O(n)$

C++代码：

class Solution {
    static constexpr int dir[4][2] = {{0,1},{0,-1},{1,0},{-1,0}};
public:
    bool canReach(vector<int>& arr, int start) {
        int n = arr.size();
        vector<bool> st(n);

        queue<int> q;
        q.push(start);
        //记录该点是否被访问过
        st[start] = true;

        while(!q.empty()){
            auto t = q.front();
            q.pop();
            //如果直接跳到值为 0 的点，返回 true
            if(arr[t] == 0) return true;

            int d = arr[t];
            if(t + d < n && !st[t + d]){
                st[t + d] = true;
                q.push(t + d);
            }
            if(t - d >= 0 && !st[t - d]){
                st[t - d] = true;
                q.push(t - d);
            }
        }
        return false;
    }
};