Leetcode1649-通过指令创建有序数组

最新推荐文章于 2024-05-30 20:15:33 发布

wobushihaorenF

最新推荐文章于 2024-05-30 20:15:33 发布

阅读量89

点赞数

分类专栏： Leetcode 树状数组文章标签：算法 leetcode c++

本文链接：https://blog.csdn.net/aaa7888210/article/details/128328574

版权

Leetcode 同时被 2 个专栏收录

130 篇文章 1 订阅

订阅专栏

树状数组

12 篇文章 0 订阅

订阅专栏

树状数组。

模板题，每次查询小于instructions[i]的元素或者大于等于instructions[i]的复杂度是logn级别，将instructions[i]插入到数组中也是logn级别。

class Solution {
public:
    int mod=1e9+7;
    int tr[100010];
    int n=100005;
    int lowbit(int x){
        return x&-x;
    }
    void add(int u,int k){
        for(int i=u;i<=n;i+=lowbit(i)) tr[i]+=k;
    }
    int query(int u){
        int res=0;
        for(int i=u;i;i-=lowbit(i)) res+=tr[i];
        return res;
    }
    int createSortedArray(vector<int>& instructions) {
        int ans=0;
        for(auto c:instructions){
            ans=(ans+min(query(c-1),query(n)-query(c))%mod+mod)%mod;
            add(c,1);
        }
        return ans;
    }
};

时间复杂度：O（nlogn）。

空间复杂度：O（n）。