题目:
给定一个整数数组和一个整数 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
思路:
可以采用简单的暴力解法,这里不赘述。另一种思路就是采用unorder_set
保存第i个元素钱的k个元素,每次对与nums[i]
查找再unorder_set
中是否存在该元素,若存在,则直接返回true,否则,将该元素放入unorder_set
中,同时,为了保证unorder_set
的元素个数小于等于k,所以当元素个数大于k时,删除掉unorder_set
中的第一个元素。
class Solution {
public:
bool containsNearbyDuplicate(vector<int>& nums, int k) {
unordered_set<int> record; //用于记录已经遍历过的数字
for(int i = 0; i < nums.size(); i++){
if(record.find(nums[i]) != record.end())
return true;
record.insert(nums[i]);
if(record.size() > k)
record.erase(nums[i - k]);
}
return false;
}
};