题目描述
给定一个整数数组和一个整数 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范围内的两数是否相等,超时!!
代码1
/*
暴力方法,时间复杂度为O(n^2)
*/
class Solution {
public:
bool containsNearbyDuplicate(vector<int>& nums, int k) {
int len = nums.size();//长度
bool flag = false;
for(int i = 0; i < len; ++i){
if(!flag){
for(int j = i + 1; j <= i + k && j < len; ++j){
if(nums[i] == nums[j]){
flag = true;
break;
}
}
}
}
return flag;
}
};
题解2
因为判断两个数是否相等,存在查找。如果利用set和map这些数据结构,速度肯定快很多。
具体思路为:建立一个map,key为元素值,value为下标。遍历nums数组,判断元素是否在map中存在,不存在,加入。存在,判断下标之差是否不大于k,大于,则更新当前元素的value为较大的下标,否则,则找到符合条件的两个元素,返回true。
·
代码2
class Solution {
public:
bool containsNearbyDuplicate(vector<int>& nums, int k) {
int len = nums.size();//长度
bool flag = false;
map<int, int>myMap;
for(int i = 0; i < len; ++i){
if(myMap.find(nums[i]) ==myMap.end()){//不存在
myMap[nums[i]] = i;
}
else{
int j = myMap[nums[i]];
if(i - j <= k){//符合条件
flag = true;
break;
}
else{//不符合,更新下标为较大值
myMap[nums[i]] = i;
}
}
}
return flag;
}
};