LeetCode刷题（181）～存在重复元素 II【unordered_set+滑动窗口 优秀！】

题目描述

给定一个整数数组和一个整数 k，判断数组中是否存在两个不同的索引 i 和 j，使得 nums [i] = nums [j]，并且 i 和 j 的差的 绝对值 至多为 k。

示例 1:

输入: nums = [1,2,3,1], k = 3
输出: true

  

 

  
1
  
2
 

示例 2:

输入: nums = [1,0,1,1], k = 1
输出: true

  

 

  
1
  
2
 

示例 3:

输入: nums = [1,2,3,1,2,3], k = 2
输出: false

  

 

  
1
  
2
 

解答 By 海轰

提交代码（暴力：超时）

bool containsNearbyDuplicate(vector<int>& nums, int k) { for(int i=0;i<nums.size();++i){ for(int j=i+1;j<=i+k&&j<nums.size();++j){ if(nums[i]==nums[j]) return true; } } return false; }

  

 

  
1
  
2
  
3
  
4
  
5
  
6
  
7
  
8
  
9
 

运行结果

提交代码（使用unordered_set 维持一个滑动窗口）

bool containsNearbyDuplicate(vector<int>& nums, int k) { unordered_set<int> s; for(int i=0;i<nums.size();++i){ if(s.find(nums[i])!=s.end()){ return true; } s.insert(nums[i]); if(s.size()>k){ s.erase(nums[i-k]);// 注意这里是删除nums[i-k] 不是s.begin() } } return false; }

  

 

  
1
  
2
  
3
  
4
  
5
  
6
  
7
  
8
  
9
  
10
  
11
  
12
  
13
 

运行结果

题目来源

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/contains-duplicate-ii

文章来源: haihong.blog.csdn.net，作者：海轰Pro，版权归原作者所有，如需转载，请联系作者。

原文链接：haihong.blog.csdn.net/article/details/108923854