一. 题目描述
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
,当且仅当存在两个不同的下标i
和j
,满足: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笔记:Contains Duplicate II
原文:http://blog.csdn.net/liyuefeilong/article/details/50732773