给你一个整数数组 nums 和一个整数 k ,判断数组中是否存在两个 不同的索引 i 和 j ,满足 nums[i] == nums[j] 且 abs(i - j) <= k 。如果存在,返回 true ;否则,返回 false 。
示例 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
思路:
使用Hashset的特性,里面始终最多包含 k 个元素,当出现重复值时则说明在 k 距离内存在重复元素。
每次遍历一个元素则将其加入HashSet中,如果哈希表的大小大于 k,则移除最前面的数字。保持里面只有k个元素。
代码:
public class Test05 {
public static void main(String[] args) {
Test05 test = new Test05();
int[] nums = {1, 2, 2, 4, 2, 3};
int k = 2;
boolean b = test.containsNearbyDuplicate(nums, k);
System.out.println("是否存在:" + b);
}
public boolean containsNearbyDuplicate(int[] nums, int k) {
HashSet hashSet = new HashSet();
for (int i = 0; i < nums.length; i++) {
if (hashSet.contains(nums[i])) {
return true;
}
hashSet.add(nums[i]);
if (hashSet.size() > k) {
// 使用 remove() 方法移除指定元素
hashSet.remove(nums[i - k]);
}
}
return false;
}
}
结果: