解题思路-leetcode第220题:存在重复元素Ⅲ
题目描述:
在整数数组 nums 中,是否存在两个下标 i 和 j,使得 nums [i] 和 nums [j] 的差的绝对值小于等于 t ,且满足 i 和 j 的差的绝对值也小于等于 ķ 。
如果存在则返回 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
解题思路:
本题采用滑动窗口的方法,创建不重复集合set存储窗口元素,遍历数组,当t等于0时,判断当前元素是否在集合内,若在,则返回True,否则遍历窗口元素,若存在元素与当前元素的差的绝对值小于k,则返回True。最后判断集合长度,大于k,去掉集合第一个元素。代码如下:
class Solution:
def containsNearbyAlmostDuplicate(self, nums: List[int], k: int, t: int) -> bool:
n = len(nums)
if n <= 1:
return False
l = set()
for i in range(n):
if t == 0:
if nums[i] in l:
return True
else:
for j in l:
if abs(j - nums[i]) <= t:
return True
l.add(nums[i])
if len(l) > k:
l.remove(nums[i-k])
return False
提交后,通过。