给定一个整数数组和一个整数 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
C++
class Solution {
public:
bool containsNearbyDuplicate(vector<int>& nums, int k)
{
int n=nums.size();
map<int,int> dic;
for(int i=0;i<n;i++)
{
if(dic.count(nums[i])==0)
{
dic.insert(pair<int,int>(nums[i],i));
}
else
{
if(i-dic[nums[i]]<=k)
{
return true;
}
else
{
dic[nums[i]]=i;
}
}
}
return false;
}
};
python
class Solution:
def containsNearbyDuplicate(self, nums, k):
"""
:type nums: List[int]
:type k: int
:rtype: bool
"""
dic={}
n=len(nums)
for i in range(0,n):
if nums[i] not in dic:
dic[nums[i]]=i
else:
if i-dic[nums[i]]<=k:
return True
else:
dic[nums[i]]=i
return False