LeetCode 348： 存在重复元素 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

解题

和

存在重复元素I

略有不同，因为判断相同元素之间的绝对距离，因此排序法不适用，只能借助哈希表建立数值和索引之间的映射，如果当前元素已经存储到哈希表中，计算当前索引与哈希表记录的索引的距离，如果距离小于k，返回true，否则用当前索引更新哈希表中的索引，因为前面的索引值已经无效；如果当前元素第一次出现，则直接保存索引值。

class Solution {
public:
    bool containsNearbyDuplicate(vector<int>& nums, int k) {
        unordered_map<int, int> int_int;
        for (int i=0; i<nums.size(); ++i){
            if (int_int.find(nums[i]) == int_int.end())
                int_int[nums[i]] = i;
            else{
                if (i - int_int[nums[i]] <= k)
                    return true;
                else
                    int_int[nums[i]] = i;
            }
        }
        return false;
    }
};