leetcode解题之存在重复元素III

给定一个整数数组，判断数组中是否有两个不同的索引 i 和 j，使得 nums [i] 和 nums [j] 的差的绝对值最大为 t，并且 i
和 j 之间的差的绝对值最大为 ķ。

示例 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

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

有之前两个重复元素的基础，直接上手写

class Solution {
    public boolean containsNearbyAlmostDuplicate(int[] nums, int k, int t) {
        for(int i=0;i<nums.length-1;i++){
            for(int j=i+1;j<nums.length;j++){
                if(Math.abs(nums[i]-nums[j])<=t&&Math.abs(i-j)<=k){
                    return true;
                }
            }
        }
        return false;
    }
}

提交，悲剧产生了，整数溢出了，测试用例

输入：
[-1,2147483647]
1
2147483647

怎么办，将整数转化为长整型

class Solution {
    public boolean containsNearbyAlmostDuplicate(int[] nums, int k, int t) {
        for(int i=0;i<nums.length-1;i++){
            for(int j=i+1;j<nums.length;j++){
                if(Math.abs((long)nums[i]-(long)nums[j])<=t&&Math.abs(i-j)<=k){
                    return true;
                }
            }
        }
        return false;
    }
}

使用二叉树，不是太懂，记录一下

class Solution {
    public boolean containsNearbyAlmostDuplicate(int[] nums, int k, int t) {
        TreeSet<Integer> set = new TreeSet();
        for(int i=0;i<nums.length;i++){
            Integer maxNum=set.ceiling(nums[i]);
            if(maxNum!=null&&maxNum<=t+nums[i]) return true;

            Integer minNum=set.floor(nums[i]);
            if(minNum!=null&&nums[i]<=minNum+t) return true;
            set.add(nums[i]);
            if(set.size()>k){
                set.remove(nums[i-k]);
            }
            }
        return false;
    }
}

详细请查看官方题解