LeetCode 题解工作台
找出满足差值条件的下标 I
给你一个下标从 0 开始、长度为 n 的整数数组 nums ,以及整数 indexDifference 和整数 valueDifference 。 你的任务是从范围 [0, n - 1] 内找出 2 个满足下述所有条件的下标 i 和 j : abs(i - j) >= indexDifference…
2
题型
6
代码语言
3
相关题
当前训练重点
简单 · 双·指针·invariant
答案摘要
我们用两个指针 和 来维护一个间隔为 的滑动窗口,其中指针 和 分别指向窗口的左右边界。初始时 指向 ,而 指向 。 我们用 和 来维护指针 左侧区间的最小值下标和最大值下标。
Interview AiBoxInterview AiBox 实时 AI 助手,陪你讲清 双·指针·invariant 题型思路
题目描述
给你一个下标从 0 开始、长度为 n 的整数数组 nums ,以及整数 indexDifference 和整数 valueDifference 。
你的任务是从范围 [0, n - 1] 内找出 2 个满足下述所有条件的下标 i 和 j :
abs(i - j) >= indexDifference且abs(nums[i] - nums[j]) >= valueDifference
返回整数数组 answer。如果存在满足题目要求的两个下标,则 answer = [i, j] ;否则,answer = [-1, -1] 。如果存在多组可供选择的下标对,只需要返回其中任意一组即可。
注意:i 和 j 可能 相等 。
示例 1:
输入:nums = [5,1,4,1], indexDifference = 2, valueDifference = 4 输出:[0,3] 解释:在示例中,可以选择 i = 0 和 j = 3 。 abs(0 - 3) >= 2 且 abs(nums[0] - nums[3]) >= 4 。 因此,[0,3] 是一个符合题目要求的答案。 [3,0] 也是符合题目要求的答案。
示例 2:
输入:nums = [2,1], indexDifference = 0, valueDifference = 0 输出:[0,0] 解释: 在示例中,可以选择 i = 0 和 j = 0 。 abs(0 - 0) >= 0 且 abs(nums[0] - nums[0]) >= 0 。 因此,[0,0] 是一个符合题目要求的答案。 [0,1]、[1,0] 和 [1,1] 也是符合题目要求的答案。
示例 3:
输入:nums = [1,2,3], indexDifference = 2, valueDifference = 4 输出:[-1,-1] 解释:在示例中,可以证明无法找出 2 个满足所有条件的下标。 因此,返回 [-1,-1] 。
提示:
1 <= n == nums.length <= 1000 <= nums[i] <= 500 <= indexDifference <= 1000 <= valueDifference <= 50
解题思路
方法一:双指针 + 维护最大最小值
我们用两个指针 和 来维护一个间隔为 的滑动窗口,其中指针 和 分别指向窗口的左右边界。初始时 指向 ,而 指向 。
我们用 和 来维护指针 左侧区间的最小值下标和最大值下标。
当指针 向右移动时,我们需要更新 和 。如果 ,则 更新为 ;如果 ,则 更新为 。更新完 和 后,我们就可以判断是否找到了满足条件的下标对。如果 ,则找到了满足条件的下标对 ;如果 ,则找到了满足条件的下标对 。
如果指针 移动到了数组的末尾,说明没有找到满足条件的下标对,返回 。
时间复杂度 ,其中 是数组的长度。空间复杂度 。
class Solution:
def findIndices(
self, nums: List[int], indexDifference: int, valueDifference: int
) -> List[int]:
mi = mx = 0
for i in range(indexDifference, len(nums)):
j = i - indexDifference
if nums[j] < nums[mi]:
mi = j
if nums[j] > nums[mx]:
mx = j
if nums[i] - nums[mi] >= valueDifference:
return [mi, i]
if nums[mx] - nums[i] >= valueDifference:
return [mx, i]
return [-1, -1]
复杂度分析
| 指标 | 值 |
|---|---|
| 时间 | Depends on the final approach |
| 空间 | Depends on the final approach |
面试官常问的追问
外企场景- question_mark
The candidate demonstrates familiarity with both brute-force and optimized solutions.
- question_mark
The candidate considers edge cases such as arrays with small lengths or when no solution exists.
- question_mark
The candidate may discuss trade-offs between clarity and efficiency in choosing the solution.
常见陷阱
外企场景- error
Relying too heavily on the brute-force approach without considering optimization strategies like two-pointer scanning.
- error
Forgetting to handle edge cases like when the array is very small or when no valid pair can be found.
- error
Incorrectly assuming that the two-pointer approach always works for every problem; ensuring correct invariant tracking is crucial.
进阶变体
外企场景- arrow_right_alt
What if there is a constraint that both indexDifference and valueDifference are always 0?
- arrow_right_alt
How would the problem change if the array is sorted?
- arrow_right_alt
What happens if nums contains duplicate values?