leetcode笔记:Contains Duplicate II

一. 题目描述

Given an array of integers and an integer k, find out whether there are two distinct indices i and j in the array such that nums[i] = nums[j] and the difference between i and j is at most k.

二. 题目分析

题目的大意是,给定一个整数数组与一个整数k,当且仅当存在两个不同的下标ij,满足:nums[i] = nums[j],且| i - j | <= k时返回true,反之返回false

此题在前面一题Contains Duplicate的基础上,要求两个重复元素的下标相差不超过k。与上题的解法类似,令map中元素的值为key,元素下标为value,遍历数组。当找到重复元素时,判断两个元素的下标差是否小于等于k。若大于k,更新已在map中元素对应的value,即下标的值。

三. 示例代码

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

四. 小结

该题目使用unordered_map比map效率高。后续题目有:Contains Duplicate III。

你可能感兴趣的:(LeetCode,Algorithm,C++,算法,map)