No.137- LeetCode307 - 线段树模板题

原创于 2019-11-05 20:47:59 发布 · 178 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#LeetCode

leetcode 专栏收录该内容

251 篇文章

订阅专栏

本文介绍了一种使用段式树状数组实现的高效数据结构，该结构支持快速更新和区间查询操作。通过递归构建树状结构，并利用更新和查询算法，可以实现在O(log N)时间内对数组元素进行修改和查询区间和的功能。

class NumArray {
private:
    struct Node{
        int sum;
        Node* left;
        Node* right;
        Node():sum(0),left(NULL),right(NULL){}
    };
    int N;
    Node* root;
public:
    NumArray(vector<int>& nums) {
        N = nums.size();
        root = build(0,N);
        for(int i=0;i<N;i++) update(i,nums[i]);
    }

    Node* build(int L,int R){
        Node* t = new Node();
        if(L+1 >= R) return t;
        int mid = (L+R)/2;
        t->left = build(L,mid);
        t->right = build(mid,R);
        return t;
    }

    void updateA(Node* now,int loc,int L,int R,int key,int& T){
        printf("<%d %d>\n",L,R);
        if(L+1 >= R){
            T = key - now->sum;
            now->sum = key;
            return ;
        }
        int mid = (L+R)/2;
        if(loc < mid) updateA(now->left,loc,L,mid,key,T);
        else updateA(now->right,loc,mid,R,key,T);
        now->sum += T;
    }

    void update(int i, int val){
        int t;
        updateA(root,i,0,N,val,t);
    }

    int bsearch(Node* now,int l,int r,int L,int R){
        if(l==L && r==R) return now->sum;
        int mid = (L+R)/2;
        if(r <= mid) return bsearch(now->left,l,r,L,mid);
        if(l >= mid) return bsearch(now->right,l,r,mid,R);
        return bsearch(now->left,l,mid,L,mid) + bsearch(now->right,mid,r,mid,R);
    }

    int sumRange(int i, int j) {return bsearch(root,i,j+1,0,N);}
};