Offer 驾到,掘友接招!我正在参与2022春招打卡活动,点击查看活动详情。
一、题目描述
219. 存在重复元素 II - 力扣(LeetCode) (leetcode-cn.com)
给你一个整数数组 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 <= 10^5
- -109 <= nums[i] <= 10^9
- 0 <= k <= 105
二、思路分析
通过set来记录在[l...l+k]这个区间内的元素,如果范围超出了,就去掉当前索引-k这一个元素,也就是区间的第一个元素,从而保证了滑动窗口的的大小固定
三、AC 代码
class Solution {
public boolean containsNearbyDuplicate(int[] nums, int k) {
int n = nums.length;
Set<Integer> set = new HashSet<>();
for( int i = 0; i < n; i ++){
if(set.contains(nums[i]))
return true;
set.add(nums[i]);
if(set.size() == k + 1)
set.remove(nums[i-k]);
}
return false;
}
}
四、总结
这个问题用 set 可以更快,因为字典查找是 log(n) 级别。
范文参考
画解算法:219. 存在重复元素 II - 存在重复元素 II - 力扣(LeetCode) (leetcode-cn.com)
【i和j的差的绝对值最多为k】->window=【i,i+k】 - 存在重复元素 II - 力扣(LeetCode) (leetcode-cn.com)