[LeetCode] 219. 存在重复元素 II

福楠2024-10-14 3:09

题目描述：

给你一个整数数组 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

题目链接：

. - 力扣（LeetCode）

解题主要思想：

其实这道题跟 "两数之和" 有点像，但需要注意的是，题目说的是<= k，因此我们需要尽可能地缩小i和j的距离，但如果是>= k，那我们就需要尽可能地增大i和j的距离，解决办法就是只记录第一次发现nums $i$ 时的下标。

两数之和链接：
$LeetCode$ 1. 两数之和-CSDN博客

解题代码：

cpp 复制代码

class Solution {
public:
    bool containsNearbyDuplicate(vector<int>& nums, int k) {
        unordered_map<int, int> map;
        for(int i = 0; i < nums.size(); ++i) {
            if (map.count(nums[i])) {
                if ((i-map[nums[i]]) <= k) return true;
            }
            map[nums[i]] = i;  // 如果是>= k，那么就需要加上在这行代码最前面加上else
        }
        return false;
    }
};