线段树--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!

发布了51 篇原创文章 · 获赞 6 · 访问量 2万+

猜你喜欢

转载自blog.csdn.net/qq_35776579/article/details/53204666