题目地址:
https://leetcode.com/problems/contains-duplicate-ii/
给定一个长 n n n的数组 A A A,问是否存在数组中两个相同的数其下标差小于等于 k k k。
用一个哈希表记录值及其最新出现的下标。每次遍历到 A [ i ] A[i] A[i]的时候,先查一下哈希表里是否存在 A [ i ] A[i] A[i]的key,并且其下标与 i i i的差距不超过 k k k,如果存在则返回true。代码如下:
class Solution {
public:
bool containsNearbyDuplicate(vector<int>& A, int k) {
unordered_map<int, int> mp;
for (int i = 0; i < A.size(); i++) {
if (mp.count(A[i]) && i - mp[A[i]] <= k) return true;
mp[A[i]] = i;
}
return false;
}
};
时空复杂度 O ( n ) O(n) O(n)。