Contains Duplicate II
leetcode上第219号问题:Contains Duplicate II
给出⼀个整形数组nums和⼀个整数k,是否存在索引i和j,使得nums[i] == nums[j] 且i和j之间的差不超过k Example 1: Input: nums = [1,2,3,1], k = 3 Output: true. Example 2: Input: nums = [1,0,1,1], k = 1 Output: true Example 3: Input: nums = [1,2,3,1,2,3], k = 2 Output: false
考虑用滑动窗口与查找表来解决。
record,用来保存每次遍历时插入的元素,record的最大长度为knums,每次遍历的时候在record查找是否存在相同的元素,如果存在则返回true,遍历结束record未查找到,则将该元素插入到record中,而后查看record的长度是否为k + 1record的长度是否为k + 1,则删减record的元素,该元素的值为nums[i - k]nums未查找到则返回false动画演示
1// 219. Contains Duplicate II
2// https://leetcode.com/problems/contains-duplicate-ii/description/
3// 时间复杂度: O(n)
4// 空间复杂度: O(k)
5class Solution {
6public:
7 bool containsNearbyDuplicate(vector<int>& nums, int k) {
8
9 if(nums.size() <= 1) return false;
10
11 if(k <= 0) return false;
12
13
14 unordered_set<int> record;
15 for(int i = 0 ; i < nums.size() ; i ++){
16
17 if(record.find(nums[i]) != record.end()){
18 return true;
19 }
20
21 record.insert(nums[i]);
22
23 // 保持record中最多有k个元素
24 // 因为在下一次循环中会添加一个新元素,使得总共考虑k+1个元素
25 if(record.size() == k + 1){
26 record.erase(nums[i - k]);
27 }
28 }
29
30 return false;
31 }
32};执行结果