Off-by-one on range boundaries
Wrong move: Loop endpoints miss first/last candidate.
Usually fails on: Fails on minimal arrays and exact-boundary answers.
Fix: Re-derive loops from inclusive/exclusive ranges before coding.
Break down a hard problem into reliable checkpoints, edge-case handling, and complexity trade-offs.
You are given an integer n representing the number of nodes in a graph, labeled from 0 to n - 1.
You are also given an integer array nums of length n and an integer maxDiff.
An undirected edge exists between nodes i and j if the absolute difference between nums[i] and nums[j] is at most maxDiff (i.e., |nums[i] - nums[j]| <= maxDiff).
You are also given a 2D integer array queries. For each queries[i] = [ui, vi], find the minimum distance between nodes ui and vi. If no path exists between the two nodes, return -1 for that query.
Return an array answer, where answer[i] is the result of the ith query.
Note: The edges between the nodes are unweighted.
Example 1:
Input: n = 5, nums = [1,8,3,4,2], maxDiff = 3, queries = [[0,3],[2,4]]
Output: [1,1]
Explanation:
The resulting graph is:
| Query | Shortest Path | Minimum Distance |
|---|---|---|
| [0, 3] | 0 → 3 | 1 |
| [2, 4] | 2 → 4 | 1 |
Thus, the output is [1, 1].
Example 2:
Input: n = 5, nums = [5,3,1,9,10], maxDiff = 2, queries = [[0,1],[0,2],[2,3],[4,3]]
Output: [1,2,-1,1]
Explanation:
The resulting graph is:
| Query | Shortest Path | Minimum Distance |
|---|---|---|
| [0, 1] | 0 → 1 | 1 |
| [0, 2] | 0 → 1 → 2 | 2 |
| [2, 3] | None | -1 |
| [4, 3] | 3 → 4 | 1 |
Thus, the output is [1, 2, -1, 1].
Example 3:
Input: n = 3, nums = [3,6,1], maxDiff = 1, queries = [[0,0],[0,1],[1,2]]
Output: [0,-1,-1]
Explanation:
There are no edges between any two nodes because:
|nums[0] - nums[1]| = |3 - 6| = 3 > 1|nums[0] - nums[2]| = |3 - 1| = 2 > 1|nums[1] - nums[2]| = |6 - 1| = 5 > 1Thus, no node can reach any other node, and the output is [0, -1, -1].
Constraints:
1 <= n == nums.length <= 1050 <= nums[i] <= 1050 <= maxDiff <= 1051 <= queries.length <= 105queries[i] == [ui, vi]0 <= ui, vi < nProblem summary: You are given an integer n representing the number of nodes in a graph, labeled from 0 to n - 1. You are also given an integer array nums of length n and an integer maxDiff. An undirected edge exists between nodes i and j if the absolute difference between nums[i] and nums[j] is at most maxDiff (i.e., |nums[i] - nums[j]| <= maxDiff). You are also given a 2D integer array queries. For each queries[i] = [ui, vi], find the minimum distance between nodes ui and vi. If no path exists between the two nodes, return -1 for that query. Return an array answer, where answer[i] is the result of the ith query. Note: The edges between the nodes are unweighted.
Start with the most direct exhaustive search. That gives a correctness anchor before optimizing.
Pattern signal: Array · Two Pointers · Binary Search · Dynamic Programming · Greedy · Bit Manipulation
5 [1,8,3,4,2] 3 [[0,3],[2,4]]
5 [5,3,1,9,10] 2 [[0,1],[0,2],[2,3],[4,3]]
3 [3,6,1] 1 [[0,0],[0,1],[1,2]]
Source-backed implementations are provided below for direct study and interview prep.
// Accepted solution for LeetCode #3534: Path Existence Queries in a Graph II
class Solution {
public int[] pathExistenceQueries(int n, int[] nums, int maxDiff, int[][] queries) {
int[] ans = new int[queries.length];
int[] indexMap = new int[n];
int[] sortedNums = new int[n];
Pair<Integer, Integer>[] sortedNumAndIndexes = new Pair[n];
for (int i = 0; i < n; ++i)
sortedNumAndIndexes[i] = new Pair<>(nums[i], i);
Arrays.sort(sortedNumAndIndexes, Comparator.comparingInt(Pair::getKey));
for (int i = 0; i < n; ++i) {
final int num = sortedNumAndIndexes[i].getKey();
final int sortedIndex = sortedNumAndIndexes[i].getValue();
sortedNums[i] = num;
indexMap[sortedIndex] = i;
}
final int maxLevel = Integer.SIZE - Integer.numberOfLeadingZeros(n) + 1;
// jump[i][j] := the index of the j-th ancestor of i
int[][] jump = new int[n][maxLevel];
int right = 0;
for (int i = 0; i < n; ++i) {
while (right + 1 < n && sortedNums[right + 1] - sortedNums[i] <= maxDiff)
++right;
jump[i][0] = right;
}
for (int level = 1; level < maxLevel; ++level)
for (int i = 0; i < n; ++i) {
final int prevJump = jump[i][level - 1];
jump[i][level] = jump[prevJump][level - 1];
}
for (int i = 0; i < queries.length; ++i) {
final int u = queries[i][0];
final int v = queries[i][1];
final int uIndex = indexMap[u];
final int vIndex = indexMap[v];
final int start = Math.min(uIndex, vIndex);
final int end = Math.max(uIndex, vIndex);
final int res = minJumps(jump, start, end, maxLevel - 1);
ans[i] = res == Integer.MAX_VALUE ? -1 : res;
}
return ans;
}
// Returns the minimum number of jumps from `start` to `end` using binary
// lifting.
private int minJumps(int[][] jump, int start, int end, int level) {
if (start == end)
return 0;
if (jump[start][0] >= end)
return 1;
if (jump[start][level] < end)
return Integer.MAX_VALUE;
int j = level;
for (; j >= 0; --j)
if (jump[start][j] < end)
break;
return (1 << j) + minJumps(jump, jump[start][j], end, j);
}
}
// Accepted solution for LeetCode #3534: Path Existence Queries in a Graph II
package main
import (
"math/bits"
"slices"
)
// https://space.bilibili.com/206214
func pathExistenceQueries(n int, nums []int, maxDiff int, queries [][]int) []int {
idx := make([]int, n)
for i := range idx {
idx[i] = i
}
slices.SortFunc(idx, func(i, j int) int { return nums[i] - nums[j] })
// rank[i] 表示 nums[i] 是 nums 中的第几小,或者说节点 i 在 idx 中的下标
rank := make([]int, n)
for i, j := range idx {
rank[j] = i
}
// 双指针,从第 i 小的数开始,向左一步,最远能跳到第 left 小的数
pa := make([][]int, n)
mx := bits.Len(uint(n))
left := 0
for i, j := range idx {
for nums[j]-nums[idx[left]] > maxDiff {
left++
}
pa[i] = make([]int, mx)
pa[i][0] = left
}
// 倍增
for i := range mx - 1 {
for x := range pa {
p := pa[x][i]
pa[x][i+1] = pa[p][i]
}
}
ans := make([]int, len(queries))
for qi, q := range queries {
l, r := q[0], q[1]
if l == r { // 不用跳
continue
}
l, r = rank[l], rank[r]
if l > r { // 保证 l 在 r 左边
l, r = r, l
}
// 从 r 开始,向左跳到 l
res := 0
for k := mx - 1; k >= 0; k-- {
if pa[r][k] > l {
res |= 1 << k
r = pa[r][k]
}
}
if pa[r][0] > l { // 无法跳到 l
ans[qi] = -1
} else {
ans[qi] = res + 1 // 再跳一步就能到 l
}
}
return ans
}
# Accepted solution for LeetCode #3534: Path Existence Queries in a Graph II
class Solution:
def pathExistenceQueries(
self,
n: int,
nums: list[int],
maxDiff: int,
queries: list[list[int]],
) -> list[int]:
sortedNumAndIndexes = sorted((num, i) for i, num in enumerate(nums))
sortedNums = [num for num, _ in sortedNumAndIndexes]
indexMap = {originalIndex: sortedIndex for sortedIndex,
(_, originalIndex) in enumerate(sortedNumAndIndexes)}
maxLevel = n.bit_length() + 1
# jump[i][j] is the index of the j-th ancestor of i
jump = [[0] * maxLevel for _ in range(n)]
right = 0
for i in range(n):
while right + 1 < n and sortedNums[right + 1] - sortedNums[i] <= maxDiff:
right += 1
jump[i][0] = right
for level in range(1, maxLevel):
for i in range(n):
prevJump = jump[i][level - 1]
jump[i][level] = jump[prevJump][level - 1]
def minJumps(start: int, end: int, level: int) -> int:
"""
Returns the minimum number of jumps from `start` to `end` using binary
lifting.
"""
if start == end:
return 0
if jump[start][0] >= end:
return 1
if jump[start][level] < end:
return math.inf
for j in range(level, -1, -1):
if jump[start][j] < end:
break
return (1 << j) + minJumps(jump[start][j], end, j)
def minDist(u: int, v: int) -> int:
uIndex = indexMap[u]
vIndex = indexMap[v]
start = min(uIndex, vIndex)
end = max(uIndex, vIndex)
res = minJumps(start, end, maxLevel - 1)
return res if res < math.inf else -1
return [minDist(u, v) for u, v in queries]
// Accepted solution for LeetCode #3534: Path Existence Queries in a Graph II
// Rust example auto-generated from java reference.
// Replace the signature and local types with the exact LeetCode harness for this problem.
impl Solution {
pub fn rust_example() {
// Port the logic from the reference block below.
}
}
// Reference (java):
// // Accepted solution for LeetCode #3534: Path Existence Queries in a Graph II
// class Solution {
// public int[] pathExistenceQueries(int n, int[] nums, int maxDiff, int[][] queries) {
// int[] ans = new int[queries.length];
// int[] indexMap = new int[n];
// int[] sortedNums = new int[n];
// Pair<Integer, Integer>[] sortedNumAndIndexes = new Pair[n];
//
// for (int i = 0; i < n; ++i)
// sortedNumAndIndexes[i] = new Pair<>(nums[i], i);
//
// Arrays.sort(sortedNumAndIndexes, Comparator.comparingInt(Pair::getKey));
//
// for (int i = 0; i < n; ++i) {
// final int num = sortedNumAndIndexes[i].getKey();
// final int sortedIndex = sortedNumAndIndexes[i].getValue();
// sortedNums[i] = num;
// indexMap[sortedIndex] = i;
// }
//
// final int maxLevel = Integer.SIZE - Integer.numberOfLeadingZeros(n) + 1;
// // jump[i][j] := the index of the j-th ancestor of i
// int[][] jump = new int[n][maxLevel];
//
// int right = 0;
// for (int i = 0; i < n; ++i) {
// while (right + 1 < n && sortedNums[right + 1] - sortedNums[i] <= maxDiff)
// ++right;
// jump[i][0] = right;
// }
//
// for (int level = 1; level < maxLevel; ++level)
// for (int i = 0; i < n; ++i) {
// final int prevJump = jump[i][level - 1];
// jump[i][level] = jump[prevJump][level - 1];
// }
//
// for (int i = 0; i < queries.length; ++i) {
// final int u = queries[i][0];
// final int v = queries[i][1];
// final int uIndex = indexMap[u];
// final int vIndex = indexMap[v];
// final int start = Math.min(uIndex, vIndex);
// final int end = Math.max(uIndex, vIndex);
// final int res = minJumps(jump, start, end, maxLevel - 1);
// ans[i] = res == Integer.MAX_VALUE ? -1 : res;
// }
//
// return ans;
// }
//
// // Returns the minimum number of jumps from `start` to `end` using binary
// // lifting.
// private int minJumps(int[][] jump, int start, int end, int level) {
// if (start == end)
// return 0;
// if (jump[start][0] >= end)
// return 1;
// if (jump[start][level] < end)
// return Integer.MAX_VALUE;
// int j = level;
// for (; j >= 0; --j)
// if (jump[start][j] < end)
// break;
// return (1 << j) + minJumps(jump, jump[start][j], end, j);
// }
// }
// Accepted solution for LeetCode #3534: Path Existence Queries in a Graph II
// Auto-generated TypeScript example from java.
function exampleSolution(): void {
}
// Reference (java):
// // Accepted solution for LeetCode #3534: Path Existence Queries in a Graph II
// class Solution {
// public int[] pathExistenceQueries(int n, int[] nums, int maxDiff, int[][] queries) {
// int[] ans = new int[queries.length];
// int[] indexMap = new int[n];
// int[] sortedNums = new int[n];
// Pair<Integer, Integer>[] sortedNumAndIndexes = new Pair[n];
//
// for (int i = 0; i < n; ++i)
// sortedNumAndIndexes[i] = new Pair<>(nums[i], i);
//
// Arrays.sort(sortedNumAndIndexes, Comparator.comparingInt(Pair::getKey));
//
// for (int i = 0; i < n; ++i) {
// final int num = sortedNumAndIndexes[i].getKey();
// final int sortedIndex = sortedNumAndIndexes[i].getValue();
// sortedNums[i] = num;
// indexMap[sortedIndex] = i;
// }
//
// final int maxLevel = Integer.SIZE - Integer.numberOfLeadingZeros(n) + 1;
// // jump[i][j] := the index of the j-th ancestor of i
// int[][] jump = new int[n][maxLevel];
//
// int right = 0;
// for (int i = 0; i < n; ++i) {
// while (right + 1 < n && sortedNums[right + 1] - sortedNums[i] <= maxDiff)
// ++right;
// jump[i][0] = right;
// }
//
// for (int level = 1; level < maxLevel; ++level)
// for (int i = 0; i < n; ++i) {
// final int prevJump = jump[i][level - 1];
// jump[i][level] = jump[prevJump][level - 1];
// }
//
// for (int i = 0; i < queries.length; ++i) {
// final int u = queries[i][0];
// final int v = queries[i][1];
// final int uIndex = indexMap[u];
// final int vIndex = indexMap[v];
// final int start = Math.min(uIndex, vIndex);
// final int end = Math.max(uIndex, vIndex);
// final int res = minJumps(jump, start, end, maxLevel - 1);
// ans[i] = res == Integer.MAX_VALUE ? -1 : res;
// }
//
// return ans;
// }
//
// // Returns the minimum number of jumps from `start` to `end` using binary
// // lifting.
// private int minJumps(int[][] jump, int start, int end, int level) {
// if (start == end)
// return 0;
// if (jump[start][0] >= end)
// return 1;
// if (jump[start][level] < end)
// return Integer.MAX_VALUE;
// int j = level;
// for (; j >= 0; --j)
// if (jump[start][j] < end)
// break;
// return (1 << j) + minJumps(jump, jump[start][j], end, j);
// }
// }
Use this to step through a reusable interview workflow for this problem.
Two nested loops check every pair of elements. The outer loop picks one element, the inner loop scans the rest. For n elements that is n × (n−1)/2 comparisons = O(n²). No extra memory — just two loop variables.
Each pointer traverses the array at most once. With two pointers moving inward (or both moving right), the total number of steps is bounded by n. Each comparison is O(1), giving O(n) overall. No auxiliary data structures are needed — just two index variables.
Review these before coding to avoid predictable interview regressions.
Wrong move: Loop endpoints miss first/last candidate.
Usually fails on: Fails on minimal arrays and exact-boundary answers.
Fix: Re-derive loops from inclusive/exclusive ranges before coding.
Wrong move: Advancing both pointers shrinks the search space too aggressively and skips candidates.
Usually fails on: A valid pair can be skipped when only one side should move.
Fix: Move exactly one pointer per decision branch based on invariant.
Wrong move: Setting `lo = mid` or `hi = mid` can stall and create an infinite loop.
Usually fails on: Two-element ranges never converge.
Fix: Use `lo = mid + 1` or `hi = mid - 1` where appropriate.
Wrong move: An incomplete state merges distinct subproblems and caches incorrect answers.
Usually fails on: Correctness breaks on cases that differ only in hidden state.
Fix: Define state so each unique subproblem maps to one DP cell.