相比于2,增加了一个条件,即判断k距离内是否存在两个数差距在t以内。
最基本的想法,比如每次遍历一遍大小为k的set等等得到的时间代价是O(nt)或者是O(nk)的,在t或者是k较大时,效果不好。
看了一下别人的思路,很基本的做法就是分块。将(nums[i]-Math.Value_MIN)分到相应块内,保证每一块都是t+1大小的(所以每一块内各个数的最大间距为t,块内肯定满足题目要求),;将nums[i]和对应块的编号存在map中。当遇到每一个新的数nums[j]时,查看其所在组是否在map中存在(或者前后两组存在且与nums[i]的差距小于t)来判定是否满足题目条件。若不满足,插入或更新nums[j]及其所在块到map中。当map的大小超过k时,将nums[i-k]及其所在块的信息从map中删除。
public class Solution {
public boolean containsNearbyAlmostDuplicate(int[] nums, int k, int t) {
if (k < 1 || t < 0) return false;
Map<Long, Long> map = new HashMap<>();
for (int i = 0; i < nums.length; i++) {
long remappedNum = (long) nums[i] - Integer.MIN_VALUE;
long bucket = remappedNum / ((long) t+1);
if (map.containsKey(bucket)
|| (map.containsKey(bucket - 1) && remappedNum - map.get(bucket - 1) <= t)
|| (map.containsKey(bucket + 1) && map.get(bucket + 1) - remappedNum <= t))
return true;
if (map.entrySet().size() >= k) {
long lastBucket = ((long) nums[i - k] - Integer.MIN_VALUE) / ((long) t+1);
map.remove(lastBucket);
}
map.put(bucket, remappedNum);
}
return false;
}
}