存在重复元素

存在重复元素

题目出处：https://leetcode-cn.com/problems/contains-duplicate/

class Solution {
public:
    bool containsDuplicate(vector<int>& nums) {
        unordered_set<int> s;
        for (int x: nums) {
            if (s.find(x) != s.end()) {
                return true;
            }
            s.insert(x);
        }
        return false;
    }
};

存在重复元素II

题目出处：https://leetcode-cn.com/problems/contains-duplicate-ii/

思路：使用滑动窗口和有序集合的方法。维护一个大小为k的滑动，每次遍历元素x时，滑动窗口中包含元素x前面的最多k个元素，只要检查窗口中是否有元素的值等于x即可。

class Solution {
public:
    bool containsNearbyDuplicate(vector<int>& nums, int k) {
        set<int> st;
        for(int i = 0; i < nums.size(); ++i){
            if(st.find(nums[i]) != st.end()){
                return true;
            }
            st.insert(nums[i]);
            if(i >= k){
               st.erase(nums[i-k]);  
            } 
        }

        return false;
    }
};

 

存在重复元素III

题目出处：https://leetcode-cn.com/problems/contains-duplicate-iii/

思路：使用滑动窗口和有序集合的方法。维护一个大小为k的滑动，每次遍历元素x时，滑动窗口中包含元素x前面的最多k个元素，只需要检查窗口中是否存在元素落在区间[x-t,x+t]中即可。

数据结构的选择：滑动窗口中的元素应该由一个数据结构保存，该数据结构有任意添加和删除指定位置的功能；且数据结构中的元素是有序的，方便用二分法查找是否有元素落入区间[x-t,x+t]，因此可以使用有序集合。

有序集合中查找的具体思路：在set中查找是否有元素大于等于x-t，有则返回第一个大于等于x-t的元素a，若a小于等于x+t，则返回true。

set和map默认元素从小到大排列。

class Solution {
public:
    bool containsNearbyAlmostDuplicate(vector<int>& nums, int k, int t) {
        int n = nums.size();
        set<int> rec;
        for (int i = 0; i < n; i++) {
            auto iter = rec.lower_bound(max(nums[i], INT_MIN + t) - t);
            if (iter != rec.end() && *iter <= min(nums[i], INT_MAX - t) + t) {
                return true;
            }
            rec.insert(nums[i]);
            if (i >= k) {
                rec.erase(nums[i - k]);
            }
        }
        return false;
    }
};

函数说明：

vector<int> arr

upper_bound函数在#include<algorithm> 头文件中， lower_bound不需要头文件

1.  auto pos = lower_bound(arr.begin(), arr.end(), num)：从数组的begin位置到end-1位置二分查找第一个大于或等于num的数字，找到返回该数字的地址，不存在则返回end。通过返回的地址减去起始地址begin,得到找到数字在数组中的下标。【函数返回值是迭代器vector<int>::iterator, 指向查找出来的元素】

2.  auto pos = upper_bound(arr.begin(), arr.end(), num)：从数组的begin位置到end-1位置二分查找第一个大于num的数字，找到返回该数字的地址，不存在则返回end。通过返回的地址减去起始地址begin,得到找到数字在数组中的下标。【函数返回值是迭代器vector<int>::iterator】

3. auto pos = lower_bound(arr.begin(), arr.end(), num, greater()):从数组的begin位置到end-1位置二分查找第一个小于或等于num的数字，找到返回该数字的地址，不存在则返回end。通过返回的地址减去起始地址begin,得到找到数字在数组中的下标。【函数返回值是迭代器vector<int>::iterator】

4. auto pos = upper_bound(arr.begin(), arr.end(), num, greater()):从数组的begin位置到end-1位置二分查找第一个小于num的数字，找到返回该数字的地址，不存在则返回end。通过返回的地址减去起始地址begin,得到找到数字在数组中的下标。【函数返回值是迭代器vector<int>::iterator】

5.  set<int> rec;

set<int>::iterator it  =  rec.lower_bound(a);   //如果存在，则返回set中第一个大于等于a的元素的迭代器；如果不存在，则返回rec.end()。【其余三个函数类似】

【区别】因为set本身为有序集合，所以自带lower_bound函数，即可直接调用(rec.lower_bound(a))