leetcode 219题

Posted by franki on September 17, 2021

leetcode 219题 存在重复元素 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

思路

用散列表存放元素,如果在区间内发现有重复则返回true,假如超过了区间,则把最旧的数据删除,继续刚才的过程遍历,如果结束后没有重复元素,则返回false。

代码

// Set
/**
 * @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++) {
        const cur = nums[i];
        if (set.has(cur)) {
            return true;
        }
        set.add(cur);
        if (set.size > k) {
            set.delete(nums[i - k]);
        }
    }

    return false;
};

// Map
/**
 * @param {number[]} nums
 * @param {number} k
 * @return {boolean}
 */
var containsNearbyDuplicate = function(nums, k) {
    const map = new Map();

    for (let i=0; i<nums.length; i++) {
        const cur = nums[i];
        if (map.has(cur)) {
            return true;
        }
        map.set(cur, true);
        if (map.size > k) {
            map.delete(nums[i-k]);
        }
    }

    return false;
};

复杂度分析

  • 时间复杂度:O(N)
  • 空间复杂度:O(1)