【LeetCode-219】存在重复元素 II

问题

给定一个整数数组和一个整数 k,判断数组中是否存在两个不同的索引 i 和 j,使得 nums [i] = nums [j],并且 i 和 j 的差的绝对值最大为 k。

示例

输入: nums = [1,2,3,1], k = 3
输出: true

输入: nums = [1,0,1,1], k = 1
输出: true

输入: nums = [1,2,3,1,2,3], k = 2
输出: false

解答

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

重点思路

数字作为哈希表的键,键值存入对应数字的角标,当已存在键的时候比较两个角标,满足条件则输出true。

posted @ 2021-02-07 17:29  tmpUser  阅读(47)  评论(0编辑  收藏  举报