一、题目
给定一个整数数组和一个整数 k,判断数组中是否存在两个不同的索引 i 和 j,使得 nums [i] = nums [j],并且 i 和 j 的差的 绝对值 至多为 k。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/contains-duplicate-ii
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
二、思路
借助HashSet来实现,具体思路看代码
三、代码
class Solution {
public boolean containsNearbyDuplicate(int[] nums, int k) {
// 排除特殊情况,如果数组中的元素不够两个 或者k值小于1就返回false
if (nums.length < 2 || k < 1){
return false;
}
// 创建一个能放 k 个元素的 set
HashSet<Integer> set = new HashSet<>(k);
// 遍历数组
for (int i = 0; i <nums.length ; i++) {
// 如果当前元素包含在 长度为k的set中,说明正确
if (set.contains(nums[i])){
return true;
}else {
// 删除 set 中前面的元素
if (i > k -1){
set.remove(nums[i-k]);
}
// 将新的元素添加进 set 中
set.add(nums[i]);
}
}
return false;
}
}