LeetCode每日一题-4.17-220-存在重复元素III

题目描述

解法 滑动窗口 + 集合

保证abs(i - j)只需要使用滑动窗口即可，那么需要的就是在这个固定大小的窗口找到两个数，保证abs(num[i] - num[j] <= t)
由于数组的元素是无序的，所以如果我们每次加入一个新的元素都去全部比较一遍，是非常耗时的，所以就会想到在插入窗口的时候把它变成有序的。
那么理所当然就会使用集合来插入。再结合stl的set内置的api：

s.lower_bound(num)//可以找到第一个比num大的元素

就能得到代码，详细思路写在注释了。

class Solution {
//滑动窗口保证下标
//问题在于怎么找到两个相差小于t的数字
//这里使用集合，集合的Lower_bound(num)可以找到第一个比num大的元素。
//我们寻找对应数字的Lower_bound(num - t)查找比num-t大的第一个数，再看它是不是小于num+t即可
public:
    bool containsNearbyAlmostDuplicate(vector<int>& nums, int k, int t) {
        int len = nums.size();
        set<long> s;
        for(int i = 0; i < nums.size(); i++) {
            //由于-t可能小于int下限，所以改成用long
            auto cur = s.lower_bound(long(nums[i]) - t);
            if(cur != s.end() && *cur <= long(nums[i]) + t) {
                return true;
            }
            s.insert(nums[i]);
            //比如k=3，集合里面保证只有三个数字
            //那么新加入一个数字的时候，下标之差肯定<=3
            if(s.size() > k) {
                s.erase(nums[i - k]);
            }
        }
        return false;     
    }
};

时间复杂度O(nlogn)
空间复杂度O(k)

这题还有一个从结果来看比这个还快而且更简单的解法
把数组元素和下标组合成pair，放到vector，然后sort一下，然后遍历看有没有符合的就行了。

class Solution {
public:
    bool containsNearbyAlmostDuplicate(vector<int>& nums, int k, int t) {
 
        vector<pair<long,long>>v;
        for(int i = 0;i < nums.size(); i++){
            v.push_back(make_pair(nums[i],i));
        }

        sort(v.begin(),v.end(), [](const pair<long, long> a, const pair<long, long> b) {
            return a.first < b.first;
        });


        for(int i = 0;i < v.size();i++) {
            for(int j = i + 1;j < v.size();j++) {
                if(abs(v[i].first - v[j].first) <= t
                && abs(v[i].second - v[j].second) <= k) {
                    return true;
                }
                if(abs(v[i].first - v[j].first) > t){
                    break;
                }
            }
        }
        return false;
    }
};

但是从时间复杂度来看这个还有 O(n^2 + （nlogn）),可能是案例不够好。
空间复杂度O(2n)