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

wobushihaorenF

于 2022-12-15 13:11:36 发布

阅读量149

点赞数

CC 4.0 BY-SA版权

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

本文链接：https://blog.youkuaiyun.com/aaa7888210/article/details/128328574

Leetcode 同时被 2 个专栏收录

130 篇文章

订阅专栏

树状数组

12 篇文章

订阅专栏

本文介绍了一种高效的数据结构——树状数组，通过该结构可以实现快速查询和更新操作。适用于解决涉及大量元素查询和更新的问题，如创建有序数组等。文章提供了一个具体的实现案例，并分析了其时间复杂度为 O(nlogn) 和空间复杂度为 O(n)。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

树状数组。

模板题，每次查询小于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）。