0.导语
本来想着今天做一道简单题目,却被卡了这么久;
1.题目描述
给定一个整数数组和一个整数 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来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/contains-duplicate-ii
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
2.最初的思路和问题
1. 最开始的思路是:这题特简单,建立两个for循环,第一个控制读取到外围的nums[i],第二个控制读取k个数,做一个k个数的滑动即可,代码如下:
class Solution {
public:
bool containsNearbyDuplicate(vector<int>& nums, int k) {
int n = nums.size();
for(int i=0; i<n-k; i++)
{
for(int j=i+1; j<=i+k; j++)
{
if (nums[i] == nums[j]) return true;
}
}
return false;
}
};
/*
之前一直有一个数组越界的警告
一看是 两重for循环中, i=n时,j=i+k;就会发生越界
*/
然后测试很快就打我脸了
首先是一直报数组读取越界的错误,检查后发现自己在写for循环时,第二个for循环的读取界限一直丢三落四没考虑周全,i最大设为n,j还设i+k;找半天才找到这个错误;
调整了读取界限后,测试通过后;
提交时又出错了,有一个案例报错
nums = [1,1,2,3], k = 2 预期输出为:true;
然而代码输出结果为 false,检查一看,确实代码运行是这个样子:当k=2时,两个数相邻的话会在同一个窗口内,确实找不到;于是加上特殊情况处理
if(n == 2)
{
if(nums[0] == nums[1]) return true;
else return false;
}但提交遇到另一个错误:
nums = [1,2,3,4,5,6,7,8,9,9] k=3
这和上一个问题一样,只是出现在了尾部;
这说明这种思路还是有问题的!
3.新思路
参考题解,利用哈希map解决问题;
利用 hashmap, key 为 vector 中的数, value 存放对应的下标。
遍历 vector, 若当前值已经存在于 hashmap 中,判断当前的下标与 hashmap 中记录的下标差是否不大于 k,
若是,返回 true; 否则,更新 hashmap 中该值的下标,继续下一个数。
4.代码:
class Solution {
public:
bool containsNearbyDuplicate(vector<int>& nums, int k) {
int n = nums.size(), idx = 0;
unordered_map<int, int> nmap; // key:nums[i], value:index
for (int i = 0; i < n; ++i) {
auto iter = nmap.find(nums[i]);
if (iter != nmap.end()) {
if (i - iter->second <= k) return true;
else iter->second = i;
}
else nmap[nums[i]] = i;
}
return false;
}
};
5.反思
对哈希表掌握很差,需加强
6.参考:
链接:https://leetcode-cn.com/problems/contains-duplicate-ii/solution/ha-xi-map-by-zhangyl-3/