220. 存在重复元素 III

220. 存在重复元素 III

给你一个整数数组 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

解题思路：

使用O(n)的时间复杂的度话，就需要使用桶排序，桶排序是使用空间换时间的思想，需要额外申请一个O(n)的空间复杂度的桶，具体实现看代码及注释。

代码实现：

class Solution:
    def containsNearbyAlmostDuplicate(self, nums: List[int], k: int, t: int) -> bool:
        if k <= 0 or t < 0:
            return False

        # 用字典定义一个桶的集合，里面包含了多个桶，每个桶的数据结构是列表
        buckets = {}
        # 桶的数量：定义有t + 1个桶，这么定义是为了计算方便
        buckets_num = t + 1
        n = len(nums)

        # 遍历数组nums
        for i in range(n):
            # nums中的元素放入桶的计算方式
            # index是第几个桶的序号
            index = nums[i] // buckets_num

            # 判断同一个桶中是否存在当前正在遍历的元素
            # 如果是同一个桶中的元素，是满足题意要求的条件的
            if index in buckets:
                return True
            
            # 将当前遍历的元素放入桶中
            buckets[index] = nums[i]

            # 判断当前元素nums[i]所在的桶的前一个桶中的元素和当前元素nums[i]是否满足条件
            if (index - 1) in buckets and (abs(nums[i] - buckets[index - 1])) <= t:
                return True
            
            # 判断当前元素nums[i]所在的桶的后一个桶的元素和当前元素nums[i]是否满足条件
            if (index + 1) in buckets and (abs(nums[i] - buckets[index + 1])) <= t:
                return True

            
            # 如果当前遍历的数组元素的索引i大于等于题目要求的k的时候，需要将最早被放入桶中元素的那个桶，即删除旧桶
            # 删除元素是nums[i-k]，元素所在的桶编号是nums[i-k] // buckets_num
            # 以保证上面的多个判断条件中能正常返回True，即桶中的元素索引和第i+1的差值小于等于k
            # 其实桶中只有一个元素，删除桶就是删除了一个元素
            if i >= k:
                del buckets[nums[i-k] // buckets_num]

        # 都不满足，返回False
        return False

参考文献:
https://leetcode-cn.com/problems/contains-duplicate-iii/solution/li-yong-tong-de-yuan-li-onpython3-by-zhou-pen-chen/