LeetCode 题解工作台
存在重复元素 III
给你一个整数数组 nums 和两个整数 indexDiff 和 valueDiff 。 找出满足下述条件的下标对 (i, j) : i != j , abs(i - j) abs(nums[i] - nums[j]) 如果存在,返回 true ; 否则,返回 false 。 示例 1: 输入: nu…
5
题型
6
代码语言
3
相关题
当前训练重点
困难 · 滑动窗口(状态滚动更新)
答案摘要
我们维护一个大小为 的滑动窗口,窗口中的元素保持有序。 遍历数组 `nums`,对于每个元素 ,我们在有序集合中查找第一个大于等于 $nums[i] - t$ 的元素,如果元素存在,并且该元素小于等于 $nums[i] + t$,说明找到了一对符合条件的元素,返回 `true`。否则,我们将 插入到有序集合中,并且如果有序集合的大小超过了 ,我们需要将最早加入有序集合的元素删除。
Interview AiBoxInterview AiBox 实时 AI 助手,陪你讲清 滑动窗口(状态滚动更新) 题型思路
题目描述
给你一个整数数组 nums 和两个整数 indexDiff 和 valueDiff 。
找出满足下述条件的下标对 (i, j):
i != j,abs(i - j) <= indexDiffabs(nums[i] - nums[j]) <= valueDiff
如果存在,返回 true ;否则,返回 false 。
示例 1:
输入:nums = [1,2,3,1], indexDiff = 3, valueDiff = 0 输出:true 解释:可以找出 (i, j) = (0, 3) 。 满足下述 3 个条件: i != j --> 0 != 3 abs(i - j) <= indexDiff --> abs(0 - 3) <= 3 abs(nums[i] - nums[j]) <= valueDiff --> abs(1 - 1) <= 0
示例 2:
输入:nums = [1,5,9,1,5,9], indexDiff = 2, valueDiff = 3 输出:false 解释:尝试所有可能的下标对 (i, j) ,均无法满足这 3 个条件,因此返回 false 。
提示:
2 <= nums.length <= 105-109 <= nums[i] <= 1091 <= indexDiff <= nums.length0 <= valueDiff <= 109
解题思路
方法一:滑动窗口 + 有序集合
我们维护一个大小为 的滑动窗口,窗口中的元素保持有序。
遍历数组 nums,对于每个元素 ,我们在有序集合中查找第一个大于等于 的元素,如果元素存在,并且该元素小于等于 ,说明找到了一对符合条件的元素,返回 true。否则,我们将 插入到有序集合中,并且如果有序集合的大小超过了 ,我们需要将最早加入有序集合的元素删除。
时间复杂度 ,其中 是数组 nums 的长度。对于每个元素,我们需要 的时间来查找有序集合中的元素,一共有 个元素,因此总时间复杂度是 。
class Solution:
def containsNearbyAlmostDuplicate(
self, nums: List[int], indexDiff: int, valueDiff: int
) -> bool:
s = SortedSet()
for i, v in enumerate(nums):
j = s.bisect_left(v - valueDiff)
if j < len(s) and s[j] <= v + valueDiff:
return True
s.add(v)
if i >= indexDiff:
s.remove(nums[i - indexDiff])
return False
复杂度分析
| 指标 | 值 |
|---|---|
| 时间 | Depends on the final approach |
| 空间 | Depends on the final approach |
面试官常问的追问
外企场景- question_mark
Look for understanding of sliding window techniques combined with efficient state management.
- question_mark
Assess the candidate's ability to handle edge cases where no valid pair exists.
- question_mark
Evaluate the candidate's knowledge of data structures like ordered sets or bucket sorts for optimization.
常见陷阱
外企场景- error
Overlooking the need for an efficient data structure to track the sliding window.
- error
Failing to properly manage the constraints, especially when dealing with large inputs.
- error
Incorrectly handling edge cases, such as when the valueDiff is zero or when no valid pair exists.
进阶变体
外企场景- arrow_right_alt
Allow a larger range for indexDiff and valueDiff, making the sliding window approach more complex.
- arrow_right_alt
Limit the array length, requiring optimization of both space and time complexity.
- arrow_right_alt
Implement the solution without relying on specific data structures like ordered sets, which may require a brute-force approach.