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
解析
解法一,DFS
我们直接定义一个访问标识数组,当前下标越界就返回false;元素已经访问过也返回false,因为再继续下去就进入了循环。然后我们再去遍历+val和-val的位置。
class Solution {
public boolean canReach(int[] arr, int start) {
boolean[] visited = new boolean[arr.length];
return dfs(arr, start, visited);
}
private boolean dfs(int[] arr, int curPos, boolean[] visited) {
if (curPos < 0 || curPos >= arr.length || visited[curPos]) return false;
if (arr[curPos] == 0) return true;
visited[curPos] = true;
return dfs(arr, curPos - arr[curPos], visited) || dfs(arr, curPos + arr[curPos], visited);
}
}
解法二,BFS
我们直接使用一个队列去存储要遍历的元素,如果元素值为0就返回。
入队逻辑是要求不越界且没有被访问过。
class Solution {
public boolean canReach(int[] arr, int start) {
LinkedList<Integer> queue = new LinkedList<>();
int n = arr.length;
boolean[] visited = new boolean[n];
queue.add(start);
while (!queue.isEmpty()) {
int size = queue.size();
for (int i = 0; i < size; i++) {
int curPos = queue.removeFirst();
int curValue = arr[curPos];
if (curValue == 0) return true;
int leftPos = curPos - curValue;
if (leftPos >= 0 && !visited[leftPos]) {
visited[leftPos] = true;
queue.addFirst(leftPos);
}
int rightPos = curPos + curValue;
if (rightPos < n && !visited[rightPos]) {
visited[rightPos] = true;
queue.addFirst(rightPos);
}
}
}
return false;
}
}
注意:本文归作者所有,未经作者允许,不得转载