题目及要求
给你一个整数数组 nums 和一个整数 k ,判断数组中是否存在两个 不同的索引 i 和 j ,满足 nums[i] == nums[j] 且 abs(i - j) <= k 。如果存在,返回 true ;否则,返回 false 。
示例 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
暴力算法
思路:遇题先来一波暴力试试能不能过,不能过的话再说,很显然这道题会超时
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<n;++j){
if(nums[i]==nums[j]&&abs(i-j)<=k){
return true;
}
}
}
return false;
}
};
哈希算法
思路:创建哈希表,用count进行查找是否存在这个数,如果不存在则将这个数存入哈希表,如果存在并且满足两个元素在数组中的距离则返回true,若循环结束都没有找到则认为不存在,返回false
时间复杂度为:O(n)
空间复杂度为:O(n)
这题的话可以不用存储到num,运行速度会更快
class Solution {
public:
bool containsNearbyDuplicate(vector<int>& nums, int k) {
unordered_map<int, int> hash;
int n=nums.size();
for(int i=0;i<n;++i){
int num=nums[i];
if(hash.count(num)&&i-hash[num]<=k){
return true;
}
hash[num]=i;
}
return false;
}
};