题意
-
给定一个整数数组和一个整数 k,判断数组中是否存在两个不同的索引 i 和 j,使得 nums [i] = nums [j],并且 i 和j
的差的绝对值最大为 k。 -
示例 1:
输入: nums = [1,2,3,1], k = 3
输出: true -
示例 2:
输入: nums = [1,0,1,1], k = 1
输出: true -
示例 3:
输入: nums = [1,2,3,1,2,3], k = 2
输出: false
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/contains-duplicate-ii
算法思路
- 维护一个size<=k的哈希表,里面存放数组里的元素。
- 遍历数组中的元素:
- 若当前元素存在于哈希表中,那么在k距离内存在相同元素,返回true
- 若不存在此元素,将其添加,然后判断此时哈希表的size是否大于k:1.大于时,移除哈希表最前面的元素,此时它已经无意义(因为在已经走过的k个距离内没有相同的元素);2.小于时,继续上面的操作。
- 若将数组元素遍历完后,没有重复元素,返回false。
代码
public boolean containsNearbyDuplicate(int[] nums, int k) {
HashSet<Integer> h=new HashSet<>();
for(int i=0;i<nums.length;i++){
if(h.contains(nums[i])){
return true;
}
h.add(nums[i]);
if(h.size()>k){
h.remove(nums[i-k]);
}
}
return false;
}