線段樹--segment_tree--lazy_tag

經過3天的努力,終於解決了lazy_tag的一系列錯誤,終於可以發篇博客了~
lazy_tag是用來區間查詢和修改的一種快速的方法,是線段樹一種比較快速的操作。(雖然有些可以用樹狀數組來解決).
至於它的思想,可以去看劉汝佳的藍書。

這裏只貼代碼~

/*
begin,end都是需要查詢的區間。
*/
#define lt (node<<1)
#define rt (node<<1|1)
struct segment_tree{
    int tree[N],lazy[N],a[N];

    void create_tree(int node,int l,int r){
        if(l==r){
            tree[node]=a[l];
            return ;
        }
        int mid=(l+r)>>1;
        create_tree(lt,l,mid);
        create_tree(rt,mid+1,r);
        tree[node]=tree[lt]+tree[rt];
        return ;
    }
/*lazy_tag操作,每次查詢到那個節點,就要把的信息傳給自己的子節點*/
    void down(int node,int l,int r){
        if(!lazy[node]) return;
        int mid=(l+r)>>1;
        tree[lt]+=(mid-l+1)*lazy[node];
        lazy[lt]+=lazy[node];
        tree[rt]+=(r-mid)*lazy[node];
        lazy[rt]+=lazy[node];
        lazy[node]=0;
    }

    int query(int node,int l,int r,int begin,int end){
        if(l>=begin && r<=end){
            return tree[node];
        }
        down(node,l,r);//每次查到這,就得將信息向下傳
        int mid=(l+r)>>1,sb=0;
        if(begin<=mid)sb+=query(lt,l,mid,begin,end);
        if(end>mid)sb+=query(rt,mid+1,r,begin,end);
        return sb;
    }

    void update(int node,int l,int r,int begin,int end,int add){
        if(l>=begin && r<=end){
            tree[node]+=(r-l+1)*add;
            lazy[node]+=add;
            return ;
        }
        int mid=(l+r)>>1;
        if(begin<=mid) update(lt,l,mid,begin,end,add);
        if(end>mid) update(rt,mid+1,r,begin,end,add);
        tree[node]+=(min(r,end)-max(l,begin)+1)*add;
    }

}d;

good luck to you!

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章