Contains Duplicate III

最新推荐文章于 2019-05-01 19:22:10 发布

最新推荐文章于 2019-05-01 19:22:10 发布 · 111 阅读

leetcode 专栏收录该内容

193 篇文章

订阅专栏

本文介绍了一种使用TreeSet在整数数组中查找是否存在两个不同的索引i和j的方法，使得nums[i]与nums[j]之间的差的绝对值不超过t，并且i与j之间的差的绝对值不超过k。

Given an array of integers, find out whether there are two distinct indices i and j in the array such that the difference between nums[i] and nums[j] is at most t and the difference between i and j is at most k.

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