力扣220题
给你一个整数数组 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
本题就是用一个长度为K的滑动窗口,每次滑动一位,从nums的0一直到nums.length-k。
在滑动窗口的每一个滑动的状态,判断该状态下的窗口里面的数是否有:
abs(nums[i] - nums[j]) <= t 。如果有,返回true结束滑动,否则一直滑动下去。如果到最后
都没有找到,则返回false。
以上这题是暴力结题的思路,提交会超时。
所有我们需要优化
这里官方通过treeset进行优化
class Solution {
public boolean containsNearbyAlmostDuplicate(int[] nums, int k, int t) {
int n = nums.length;
TreeSet<Long> set = new TreeSet<Long>();
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;
}
}
作者:LeetCode-Solution
链接:https://leetcode-cn.com/problems/contains-duplicate-iii/solution/cun-zai-zhong-fu-yuan-su-iii-by-leetcode-bbkt/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
第二种,利用hashMap的结构和桶排序的思想
class Solution {
public boolean containsNearbyAlmostDuplicate(int[] nums, int k, int t) {
int n = nums.length;
int size = t+1;
Map<Long,Long> map = new HashMap<Long,Long>();
for(int i=0;i<n;i++){
long id = getID(nums[i],size);
if(map.containsKey(id)) return true;
if(map.containsKey(id-1)&&nums[i]-map.get(id-1)<=t) return true;
if(map.containsKey(id+1)&&map.get(id+1)-nums[i]<=t) return true;
map.put(id,nums[i]*1L);
if(i>=k){
map.remove(getID(nums[i-k],size));
}
}
return false;
}
public long getID(int num,int size){
return num>=0?num/size:(num+1)/size-1;
}
}
这里面有一个步骤,删除滑动窗口之外的数,即remove第i-k个数;这里我看到有些小白会产生疑惑,
不论是
set.remove((long) nums[i - k]);
还是
map.remove(getID(nums[i-k],size));
是不是删除的不是超出长度为K的滑动窗口的数,而是滑动窗口内的数内?
其实不会,因为不论是第一种方法,使用TreeSet;还是第二种方法,使用Map。
如果滑动窗口里面的数和num[i-k]相等或者在同一个桶中,都会在这步之前的代码中,
判断存在要求的数出现,返回true了。