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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/contains-duplicate-ii/
 

求解

    class Solution {
    public:
        // 方法一，使用hash表记录相同值得所有下标
        bool containsNearbyDuplicate_1e(vector<int> &nums, int k) {
            const int n = nums.size();
            std::unordered_map<int, vector<int>> record; // <num, num相同的所有下标>
            for (int i = 0; i < n; ++i) {
                // 记录中已有和nums[i]相等的值，看是否满足条件
                if (record.count(nums[i]) > 0) {
                    auto &group = record[nums[i]];
                    for (const auto index : group) {
                        if (abs(i - index) <= k) {
                            return true;
                        }
                    }
                }
                // 注意不管是否存在和nums[i]相等的值，当前值nums[i]和下标都要更新到记录中
                record[nums[i]].emplace_back(i);
            }
            return false;
        }

        // 方法二，方法一优化，其实没必要记录所有下标，记录当前最新的下标即可，因为我们是递增遍历，下标越来越大
        bool containsNearbyDuplicate_2e(vector<int> &nums, int k) {
            const int n = nums.size();
            std::unordered_map<int, int> record; // <num, num相同的所有下标>
            for (int i = 0; i < n; ++i) {
                if (record.count(nums[i]) > 0) {
                    if (i - record[nums[i]] <= k) {
                        return true;
                    }
                }
                // 不管是否存在和nums[i]一致的值，都需要更新，除非在上面已经返回true
                record[nums[i]] = i;
            }
            return false;
        }

        // 方法三，hash表中最多只保留k个元素，在递增的过程中如果一个新元素在里面能找到，则肯定满足条件
        bool containsNearbyDuplicate(vector<int> &nums, int k) {
            const int n = nums.size();
            std::unordered_set<int> record;
            for (int i = 0; i < n; ++i) {
                int num = nums[i];
                if (record.count(num) > 0) {
                    return true;
                }
                record.emplace(num);
                // 当前元素插入后有k+1个，为了保持元素数量为k,删除最前面的加入的元素
                if (record.size() == k + 1) {
                    record.erase(nums[i - k]);
                }
            }
            return false;
        }
    };