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^4
0 <= arr[i] < arr.length
0 <= start < arr.length
思路:不要问问就是暴力,直接深搜,
建立一个vis函数标记每个已经走过的点(每个点只有两种情况,深搜的时候两个情况都会走,所以不需要二次访问),然后递归的时候注意一下i-arr[i]要>0且<arr.size()。
class Solution { public: bool canReach(vector<int>& arr, int start) { int n=arr.size(); bool ans = bfs(arr, start, n); return ans; } bool vis[500010]; bool bfs(vector<int>& arr,int start, n) { //printf("%d\n",start); if(arr[start] == 0) { return true; } else { vis[start] = true; bool flag = false; if(start + arr[start] < n && start + arr[start] >= 0 && !vis[start + arr[start]] ) flag |= bfs(arr, start + arr[start]); if(start - arr[start] < n && start - arr[start] >= 0 && !vis[start - arr[start]] ) flag |= bfs(arr, start - arr[start]); return flag; } } };