题目链接

https://leetcode-cn.com/problems/contains-duplicate-ii/

题目描述

给定一个整数数组和一个整数 k，判断数组中是否存在两个不同的索引 i 和 j，使得 nums [i] = nums [j]，并且 i 和 j 的差的绝对值最大为 k。

示例 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

解题方案

思路

• 标签：哈希

• 维护一个哈希表，里面始终最多包含k个元素，当出现重复值时则说明在k距离内存在重复元素

• 每次遍历一个元素则将其加入哈希表中，如果哈希表的大小大于k，则移除最前面的数字

• 时间复杂度：O(n)，n为数组长度

代码

• Java版本

class Solution {
   public boolean containsNearbyDuplicate(int[] nums, int k) {
       HashSet set = new HashSet<>();
       for(int i = 0; i < nums.length; i++) {
           if(set.contains(nums[i])) {
               return true;
           }
           set.add(nums[i]);
           if(set.size() > k) {
               set.remove(nums[i - k]);
           }
       }
       return false;
   }
}

• JavaScript版本

/**
* @param {number[]} nums
* @param {number} k
* @return {boolean}
*/
var containsNearbyDuplicate = function(nums, k) {
   const set = new Set();
   for(let i = 0; i < nums.length; i++) {
       if(set.has(nums[i])) {
           return true;
       }
       set.add(nums[i]);
       if(set.size > k) {
           set.delete(nums[i - k]);
       }
   }
   return false;
};

画解