剑指 Offer II 057. 值和下标之差都在给定的范围内
原始题目链接:https://leetcode.cn/problems/7WqeDu/
给你一个整数数组 nums 和两个整数 k 和 t 。请你判断是否存在 两个不同下标 i 和 j,使得 abs(nums[i] - nums[j]) <= t ,同时又满足 abs(i - j) <= k 。
如果存在则返回 true,不存在返回 false。
示例 1:
输入:nums = [1,2,3,1], k = 3, t = 0
输出:true
示例 2:
输入:nums = [1,0,1,1], k = 1, t = 2
输出:true
示例 3:
输入:nums = [1,5,9,1,5,9], k = 2, t = 3
输出:false
解题思路:
滑动窗口解题,窗口的大小是题目要求的元素下标的范围,在滑动窗口中查找一个值,满足和当前遍历的元素的差值不大于t,查找的方式有很多种,时间复杂度最小的方式是使用一个有序的数据结构作为滑动窗口存储遍历的元素,然后使用二分查找的方式找满足条件的元素。
代码实现:
class Solution:
def containsNearbyAlmostDuplicate(self, nums: List[int], k: int, t: int) -> bool:
from sortedcontainers import SortedList
# 需要一个滑动窗口,并且保持添加和删除操作后,窗口内的元素是有序的
# 所以使用SortedList()这个数据结构,便于使用二分查找算法
windows = SortedList()
for i, num in enumerate(nums):
# 题目要求是两个元素的下标索引相差k个
# 所以保持滑动窗口的元素不超过k个
windows.add(num)
# 在滑动窗口中找当前元素的索引
index = bisect_left(windows, num)
# 判断index左右两边的值与当前元素的差值是否满足小于等于t
# 右边的值
if index < len(windows) - 1 and windows[index + 1] - windows[index] <= t:
return True
if index > 0 and windows[index] - windows[index - 1] <= t:
return True
# 窗口的元素个数大于k个,需要弹出窗口中最开始进入的元素
if len(windows) > k:
windows.remove(nums[i - k])
return False
参考文献:
https://leetcode.cn/problems/7WqeDu/solution/python3-you-xu-ji-he-hua-dong-chuang-kou-6uk9/