Leetcode220——Contains Duplicate II

最新推荐文章于 2024-11-01 11:29:32 发布

cColdTea

最新推荐文章于 2024-11-01 11:29:32 发布

阅读量180

点赞数

CC 4.0 BY-SA版权

分类专栏： leetcode

本文链接：https://blog.youkuaiyun.com/emmaczw/article/details/77924799

leetcode 专栏收录该内容

16 篇文章

订阅专栏

本文介绍了一种使用滑动窗口结合多集数据结构解决数组中元素近似重复问题的方法，并提供了详细的C++代码实现。文章还探讨了另一种基于二叉搜索树的解决方案，但指出其效率较低。

很容易可以想到用一个窗口[i , i+k]，然后再在窗口内find，用mutiset这样对某个数的find的时间复杂度O(logN)，这样循环之后复杂度为(NlogN)，主要注意事项是不能用set，需要用到multiset，然后此处不能直接用find（可能find不到），用lower_bound去find就可以了

很简单，直接上代码了

class Solution {
public:
    bool containsNearbyAlmostDuplicate(vector<int>& nums, int k, int t) {
        int n = (int)nums.size();
        if(n <= 1) return false;
        multiset<long> st;
        
        for(int i = 0; i<n; i++){
            if(i-k-1 >= 0) st.erase(nums[i-k-1]);
            auto fi = st.lower_bound((long)nums[i] - t);
            if(fi != st.end() && *fi - nums[i] <= t) return true;
            st.insert((long)nums[i]);
        }
        return false;
    }
}

lz是智障，这里把insert操作写在老前面了，于是每次都是return true（自己减自己sad），蠢爆炸了。

lz是智障，这题分类是bst，于是lz用bst做了做，还是老思路，先查询并判断，然后插入，两步走，注定会超时。

struct node{
        node* l, *r;
        int num;
        int index;
        node(int num, int index): num(num), index(index), l(NULL), r(NULL) {}
    };
    
    bool Find(node* root, int num, int index, int k, int t){
        if(root == NULL) return false;
        if(index - root->index > k) return (Find(root->r, num, index, k, t) || Find(root->l, num, index, k, t));
        else{
            if(num - root->num <= t && num - root->num >= -t) return true;
            else if(num - root->num > t) return Find(root->r, num, index, k, t);
            else return Find(root->l, num, index, k, t);
        }
    }
    
    void Insert(node* &root, int num, int index){
        if(root == NULL) {root = new node(num, index); return;}
        if(num < root->num) Insert(root->l, num, index);
        else Insert(root->r, num, index);
    }
    
    bool containsNearbyAlmostDuplicate(vector<int>& nums, int k, int t) {
        int n = (int)nums.size();
        node *root = NULL;
        for(int i = 0; i<n; i++){
            if(Find(root, nums[i], i, k, t)) return true;
            else Insert(root, nums[i], i);
        }
        return false;
    }