Given an array of integers and an integer k, find out whether there are two distinct indices i and j in the array such that nums[i] = nums[j] and the difference between i and j is at most k.
给定一个整型数组和一个整数k,找出数组中是否有两个相等的整数,下标各不相同并且相差最大为k
看到这道题我首先想到的就是滑动窗口法,判断一个大小为k的窗口内的数字是否有重复的,没有再移动到下一个窗口。
于是这个方法分为几个步骤:
1. 从第一个数nums[0]开始向后往窗口添加数。窗口的整数数量不足k+1则扩大到k+1大小为止,这样第一个数和最后一个数的下标差刚好为k。
2. 对新添加进窗口的数判断HashSet中是否已经存在,是则直接返回true,否则添加到HashSet中。这里还有很多其他方法,比如用一个数组或者map对出现过的数进行标记等。
3. 窗口向后移一格,需要将被移出窗口的最左边的数从HashSet中移除。
4. 重复步骤2直到窗口移到了数组末尾。
5. 没有下标差小于等于k的相等整数的情况,最后返回false
代码如下
public class Solution {
public boolean containsNearbyDuplicate(int[] nums, int k) {
k = k + 1;
HashSet<Integer>distSet = new HashSet<Integer>();
for(int i = 0; i < nums.length;i++) {
if(i >= k) {
distSet(nums[i-k]);
}
if(distSet(nums[i])) return true;
distSet(nums[i]);
}
return false;
}
}