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∗1041 <= arr.length <= 5 * 10^41<=arr.length<=5∗104
• $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 的点，返回 trueif(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;}
};