题目描述
给你一个整数数组
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提示:
1 <= nums.length <= 105
-109 <= nums[i] <= 109
0 <= k <= 105
思路解析
把数组中每个元素,以元素:下标的形式存入map中
如果map中已经包含该元素的key,那么判断下标是否满足条件,满足条件直接返回true
如果不满足条件,那么将当前key对应值替换成当前元素下标
因为i-j>k,此时如果j不动,i继续增大,那么永远都不会满足条件
所以只有j变大才可能满足条件
代码
class Solution {public boolean containsNearbyDuplicate(int[] nums, int k) {Map<Integer, Integer> map = new HashMap<>();for (int i = 0; i < nums.length; i++) {if (!map.containsKey(nums[i])) {map.put(nums[i], i);} else {int j = map.get(nums[i]);if (i - j <= k) {return true;} else {map.put(nums[i], i);}}}// 如果没有找到重复项,则返回falsereturn false;}
}
如下是大佬思路
维护一个哈希表,里面始终最多包含 k 个元素,当出现重复值时则说明在 k 距离内存在重复元素
每次遍历一个元素则将其加入哈希表中,如果哈希表的大小大于 k,则移除最前面的数字
class Solution {public boolean containsNearbyDuplicate(int[] nums, int k) {//维护一个不包含重复元素的set,size最大为kSet<Integer> set = new HashSet<>();for (int i = 0; i < nums.length; i++) {if (!set.add(nums[i])) {// 如果添加失败(即nums[i]已经存在于set中),则返回truereturn true;}else{if(set.size()>k){//移除set首位元素,由于set没有移除首位元素方法,所以根据值来移除set.remove(nums[i-k]);}}}// 如果没有找到重复项,则返回falsereturn false;}
}