这里有一个非负整数数组 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 处。
思路:见注释
class Solution {
public:
//可以理解为二叉树的遍历,有i+arr[i]和i-arr[i]两个分叉
bool canReach(vector<int>& arr, int start) {
int size = arr.size();
//下面为二叉树bfs模板
bool inq[size] = {false};
queue<int> q;
q.push(start);
inq[start] = true;
int left,right;
while(!q.empty()){
int now = q.front();
q.pop();
if(arr[now]==0)return true;
left = now - arr[now];
right = now + arr[now];
if(left>=0 && inq[left]==false)q.push(left);
if(right<size && inq[right]==false)q.push(right);
inq[now] = true;
}
return false;
}
};