LeetCode 219. 存在重复元素 II(C++)

题目:

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

思路:

可以采用简单的暴力解法,这里不赘述。另一种思路就是采用unorder_set保存第i个元素钱的k个元素,每次对与nums[i]查找再unorder_set中是否存在该元素,若存在,则直接返回true,否则,将该元素放入unorder_set中,同时,为了保证unorder_set的元素个数小于等于k,所以当元素个数大于k时,删除掉unorder_set中的第一个元素。

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]);

            if(record.size() > k)
                record.erase(nums[i - k]);
        }
        return false;
    }
};

猜你喜欢

转载自blog.csdn.net/my_clear_mind/article/details/81783770