题解来源:画手大鹏、代码随想录
题目描述
给定一个整数数组和一个整数 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
解题思路
1.维护一个哈希表,里面始终最多包含 k 个元素,当出现重复值时则说明在 k 距离内存在重复元素
每次遍历一个元素则将其加入哈希表中,如果哈希表的大小大于 k,则移除最前面的数字
时间复杂度:O(n),n 为数组长度
class Solution {
public boolean containsNearbyDuplicate(int[] nums, int k) {
HashSet<Integer> set = new HashSet<>();
for(int i = 0; i < nums.length; i++) {
if(set.contains(nums[i])) {
return true;
}
set.add(nums[i]);
if(set.size() > k) {
set.remove(nums[i - k]);
}
}
return false;
}
}
2.创建一个map,每次操作之前,如果map里面有nums[i]相同的值,就返回true,然后更新nums[i]的最新位置,否则就加入到map中
class Solution {
public:
bool containsNearbyDuplicate(vector<int>& nums, int k) {
unordered_map<int,int> map;
for(int i = 0; i < nums.size(); i++) {
if (map.find(nums[i]) != map.end() && (i-map[nums[i]])<=k) {
return true;
}
map[nums[i]]=i;
}
return false;
}
};
注:if (map.find(nums[i]) != map.end() && (i-map[nums[i]])<=k)这句话可以改为 if(m.count(nums[i]) && i - m[nums[i]] <= k)
map.count()返回是否有这个数,返回值为1或0
map.find()返回位置
题外话
放一下我没有过的代码,时间超限。。。
class Solution {
public:
bool containsNearbyDuplicate(vector<int>& nums, int k) {
int n=nums.size();
for(int i=0;i<n;i++){
for(int j=i+1;j<=i+k&&j<n;j++){
if(nums[i]==nums[j]){
return true;
}
}
}
return false;
}
};
刷题路漫漫,艰难艰难啊~