解题思路
创建一个map,如果当前数字没有出现过,就存下这个数字和这个数字的位置
如果当前数字出现过就查看这个数字上一次出现的位置,如果二者之差小于k就返回true。
提交代码
class Solution {
public boolean containsNearbyDuplicate(int[] nums, int k) {
Map<Integer, Integer> map = new HashMap<>();
for (int i = 0; i < nums.length; i++) {
if (map.containsKey(nums[i])) {
if (i - map.get(nums[i]) <= k)
return true;
map.put(nums[i], i);
} else {
map.put(nums[i], i);
}
}
return false;
}
}
运行结果
其他解法
这道题还有一种解法,思想是滑动窗口,使用set集合来实现,思想是保持窗口的长度为k,用set来保存所有的元素,滑动窗口时,如果set集合添加元素失败说明长度为k的窗口内有重复的元素,就直接返回true:
class Solution{
public boolean containsNearbyDuplicate(int[] nums, int k) {
Set<Integer> set = new HashSet<Integer>();
for(int i = 0; i < nums.length; i++){
if(i > k) set.remove(nums[i-k-1]);
if(!set.add(nums[i])) return true;
}
return false;
}
}