题目描述
给你一个整数数组 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
初次尝试:
复杂度高,效率差。
public boolean containsNearbyDuplicate(int[] nums, int k) {
for(int i = 0; i < nums.length - 1; i++){
int j = i + 1;
int temp = k;
while(temp >= 0 && j < nums.length){
if(nums[i] == nums[j] && (Math.abs(i - j) <= k)){
return true;
}
j++;
temp--;
}
}
return false;
}
题解
滑动窗口:设置一个大小为的窗口,从0开始,窗口逐渐增大至,窗口向右滑动的时候最右侧元素下标为。当时,窗口不断增大;当时候,窗口大小为;时,需要将窗口向右滑动,也就是后,需要将下标为的元素移出窗口,并且尝试将加入窗口,如果窗口中已有该元素,则返回。
public boolean containsNearbyDuplicate(int[] nums, int k) {
//Set是元素无序且不可重复的集合
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;
}