leetcode 220. Contains Duplicate III 存在重复元素 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

提示：

0 <= nums.length <= 2 * 104
-231 <= nums[i] <= 231 - 1
0 <= k <= 104
0 <= t <= 231 - 1

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/...
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

二、解题思路

这道题关注与不同值之间的关系，有两个限制条件，两个数字的坐标差不能大于k，值差不能大于t。还是用map结构来记录值和下标的映射。这里定义两上变量i和j，开始都指向0，然后i开始向右遍历数组，如果i和j之差大于k，且map中有nums[j]，则删除并j+1，这样保证了map中所有的数的下标之差都不大于k，然后我们检查如果有值差小于等于则返回true。遍历最后返回false。

三、解题方法

3.1 Java实现

public class Solution {
    public boolean containsNearbyAlmostDuplicate(int[] nums, int k, int t) {
        int n = nums.length;
        TreeSet<Long> set = new TreeSet<>();
        for (int i = 0; i < n; i++) {
            Long ceiling = set.ceiling((long) nums[i] - (long) t);
            if (ceiling != null && ceiling <= (long) nums[i] + (long)t) {
                return true;
            }
            set.add((long) nums[i]);
            if (i >= k) {
                set.remove((long) nums[i - k]);
            }
        }
        return false;
    }
}