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;
}
};