【leetcode】220. 存在重复元素 III解题记录

题目

在整数数组 nums 中，是否存在两个下标 i 和 j，使得 nums [i] 和 nums [j] 的差的绝对值小于等于 t ，且满足 i 和 j 的差的绝对值也小于等于 ķ 。

如果存在则返回 true，不存在返回 false。

输入: nums = [1,2,3,1], k = 3, t = 0
输出: true

输入: nums = [1,0,1,1], k = 1, t = 2
输出: true

输入: nums = [1,5,9,1,5,9], k = 2, t = 3
输出: false

根据值差在t以内，想到使用桶排序

桶排序

1. 我们将数据分到 M 个桶 中。
2. 每个数字nums[i] 都被我们分配到一个桶中
3. 分配的依据就是 nums[i] // (t + 1)
4. 同一个桶内的数字最多相差t
5. 不相邻的桶一定不满足相差小于等于t

因此如果命中同一个桶内，那么直接返回True
如果命中相邻桶，我们再判断一下是否满足 相差 <= t
否则返回False
索引相差k，保持比较的桶的数量在k即可。

class Solution:
    def containsNearbyAlmostDuplicate(self, nums: List[int], k: int, t: int) -> bool:
        if t<0 :
            return False
        bucket = dict()
        for i in range(len(nums)):
            num = nums[i] // (t+1)
            if num in bucket:
                return True
            if num-1 in bucket and abs(bucket[num-1]-nums[i])<=t:
                return True
            if num+1 in bucket and abs(bucket[num+1]-nums[i])<=t:
                return True    
            bucket[num] = nums[i]
            if i>=k:
                bucket.pop(nums[i-k]//(t+1))
        return False