LeetCode 219. 存在重复元素 II

最新推荐文章于 2024-09-07 09:00:00 发布

原创最新推荐文章于 2024-09-07 09:00:00 发布 · 319 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode 专栏收录该内容

344 篇文章

订阅专栏

本文介绍了一种高效的算法，用于判断整数数组中是否存在两个不同的索引i和j，使得nums[i]=nums[j]，并且i和j的差的绝对值至多为k。通过使用unordered_map数据结构，该算法能在O(n)的时间复杂度内完成任务。

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

class Solution {
public:
    bool containsNearbyDuplicate(vector<int>& nums, int k) {
        unordered_map<int, int> val2idx;
        for (int i = 0; i < nums.size(); ++i)
        {
            if (val2idx.count(nums[i]) == 0)
                val2idx[nums[i]] = i;
            else
            {
                if (i - val2idx[nums[i]] <= k)
                    return true;
                else
                    val2idx[nums[i]] = i;
            }
        }
        return false;
    }
};