LeetCode220——存在重复元素III

我的LeetCode代码仓：https://github.com/617076674/LeetCode

原题链接：https://leetcode-cn.com/problems/contains-duplicate-iii/description/

题目描述：

知识点：二叉搜索树、滑动窗口法

思路：滑动窗口法

在LeetCode219——存在重复元素II的基础上，题目要求使得两元素差的绝对值最大为t，因此我们将hashMap改为treeMap结构存储，这样就可以用二分查找的速度来寻找到和某个新右端newNum匹配的最低值ceil。为了防止数据越界，改为Long类型来存储。

时间复杂度是O(nlogn)，其中n为数组的长度。空间复杂度是O(n)。

JAVA代码：

public class Solution {
    public boolean containsNearbyAlmostDuplicate(int[] nums, int k, int t) {
        int left = 0;
        int right = -1;			//[left, right]为我们的滑动窗口
        TreeMap<Long, Integer> treeMap = new TreeMap<>();
        while(left < nums.length) {
            if(right + 1 < nums.length && right - left < k) {
                right++;
                Long newNum = (long)nums[right];
                if(treeMap.ceilingKey(- (long)t + newNum) != null) {
                    long ceil = treeMap.ceilingKey(- (long)t + newNum);
                    if(ceil <= (long)t + newNum) {
                        return true;
                    }
                }
                if(treeMap.containsKey(newNum)) {
                    treeMap.put(newNum, treeMap.get(newNum) + 1);
                }else {
                    treeMap.put(newNum, 1);
                }
            }else {
                treeMap.put((long)nums[left], treeMap.get((long)nums[left]) - 1);
                if(treeMap.get((long)nums[left]) == 0) {
                    treeMap.remove((long)nums[left]);
                }
                left++;
            }
        }
        return false;
    }
}

LeetCode解题报告：