219. 存在重复元素 II - 力扣（LeetCode）

题目描述

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

题解1

刚开始看错题目，将“最大”二字漏掉了。然后用暴力方法判断下标之差在k范围内的两数是否相等，超时！！

代码1

/*
暴力方法，时间复杂度为O(n^2)
*/
class Solution {
public:
    bool containsNearbyDuplicate(vector<int>& nums, int k) {
        int len = nums.size();//长度
        bool flag = false;
        for(int i = 0; i < len; ++i){
            if(!flag){
                for(int j = i + 1; j <= i + k && j < len; ++j){
                    if(nums[i] == nums[j]){
                        flag = true;
                        break;
                    }
                }
            }
        }
        return flag;
    }
};

题解2

因为判断两个数是否相等，存在查找。如果利用set和map这些数据结构，速度肯定快很多。
具体思路为：建立一个map，key为元素值，value为下标。遍历nums数组，判断元素是否在map中存在，不存在，加入。存在，判断下标之差是否不大于k，大于，则更新当前元素的value为较大的下标，否则，则找到符合条件的两个元素，返回true。
·

代码2

class Solution {
public:
    bool containsNearbyDuplicate(vector<int>& nums, int k) {
        int len = nums.size();//长度
        bool flag = false;
        map<int, int>myMap;
        for(int i = 0; i < len; ++i){
            if(myMap.find(nums[i]) ==myMap.end()){//不存在
                myMap[nums[i]] = i;
            }
            else{
                int j = myMap[nums[i]];
                if(i - j <= k){//符合条件
                    flag = true;
                    break;
                }
                else{//不符合，更新下标为较大值
                    myMap[nums[i]] = i;
                }
            }
        }
        return flag;
    }
};

执行结果2