leetcode 1306. 跳跃游戏 III 中等
·
给定一个非负整数数组 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^40 <= arr[i] < arr.length0 <= 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;
}
};
更多推荐

所有评论(0)