算法学习之滑动窗口和查找表联合使用

Jackey C/C++ 1,370 次浏览 , , 没有评论

题目要求:

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

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/contains-duplicate-ii

// 时间复杂度: O(n)
// 空间复杂度: O(k)
class Solution {
public:
    bool containsNearbyDuplicate(vector<int>& nums, int k) {
        unordered_set<int> record;
        for (int i = 0; i < nums.size(); ++i) {
            if (record.find(nums[i]) != record.end())
                return true;

            record.insert(nums[i]);

            // 保证record中最多有k个元素
            if (record.size() == k + 1)
                record.erase(nums[i-k]);
        }
        return false;
    }
};

 

发表回复

您的电子邮箱地址不会被公开。 必填项已用 * 标注

Go