给定一个非负整数数组 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

分析:从 start 位置开始进行 BFS,每次检查 i + arr[i] 和 i - arr[i] 这两个位置是否已经访问过,如果没有访问过,则先标记已访问,再将这两个位置加入 BFS 队列,直到队列为空,或者访问到某个值为 0 的位置为止。

class Solution {
public:
    bool canReach(vector<int>& arr, int start) {
        int n=arr.size(),flag[n+5];
        for(int i=0;i<n;++i)
            flag[i]=0;
        flag[start]=1;
        queue<int>que;que.push(start);
        while(!que.empty())
        {
            int index=que.front(),right=index+arr[index],left=index-arr[index];que.pop();

            if(arr[index]==0)return true;
            if(right<n&&flag[right]==0)que.push(right),flag[right]=1;
            if(left>=0&&flag[left]==0)que.push(left),flag[left]=1;
        }
        return false;
    }
};

更多推荐