[CodeVS4919]线段树练习4

线段树。每个节点分别维护当前区间内%7分别等于0~6的数的个数，lazy tag记录当前区间增加的量，查询时将val“平移”lazy个即可。

#include<iostream>
#include<algorithm>
#define maxn 100001
#define root 1
#define _left <<1
#define _right <<1|1
int n;
class SegmentTree {
    private:
        int val[maxn<<2][7],lazy[maxn<<2];
        void push_up(const int p) {
            for(int i=0;i<7;i++) {
                val[p][i]=val[p _left][i]+val[p _right][i];
            }
        }
        void swap(const int p,const int q) {
            if(!lazy[p]) return;
            int t[7];
            for(int i=0;i<7;i++) t[(i+q)%7]=val[p][i];
            for(int i=0;i<7;i++) val[p][i]=t[i];
        }
        void push_down(const int p) {
            if(!lazy[p]) return;
            lazy[p _left]=lazy[p _left]+lazy[p];
            lazy[p _right]=lazy[p _right]+lazy[p];
            swap(p _left,lazy[p]);
            swap(p _right,lazy[p]);
            lazy[p]=0;
        }
    public:
        void build(const int p,const int b,const int e) {
            lazy[p]=0;
            if(b==e) {
                int x;
                std::cin>>x;
                val[p][x%7]=1;
                return;
            }
            int mid=(b+e)>>1;
            build(p _left,b,mid);
            build(p _right,mid+1,e);
            push_up(p);
        }
        void modify(const int p,const int b,const int e,const int l,const int r,const int x) {
            if(b!=e) push_down(p);
            if((b==l)&&(e==r)) {
                lazy[p]=lazy[p]+x;
                swap(p,x);
                return;
            }
            int mid=(b+e)>>1;
            if(l<=mid) modify(p _left,b,mid,l,std::min(mid,r),x);
            if(r>mid) modify(p _right,mid+1,e,std::max(mid+1,l),r,x);
            push_up(p);
        }
        int query(const int p,const int b,const int e,const int l,const int r) {
            if(b!=e) push_down(p);
            if((b==l)&&(e==r)) {
                return val[p][0];
            }
            int mid=(b+e)>>1,ans=0;
            if(l<=mid) ans+=query(p _left,b,mid,l,std::min(mid,r));
            if(r>mid) ans+=query(p _right,mid+1,e,std::max(mid+1,l),r);
            return ans;
        }
};
SegmentTree tree;
int main() {
    std::ios_base::sync_with_stdio(false);
    std::cin>>n;
    tree.build(root,1,n);
    int q;
    for(std::cin>>q;q--;) {
        char op[6];
        std::cin>>op;
        if(op[0]=='a') {
            int x,y,k;
            std::cin>>x>>y>>k;
            tree.modify(root,1,n,x,y,k);
        }
        else {
            int x,y;
            std::cin>>x>>y;
            std::cout<<tree.query(root,1,n,x,y)<<std::endl;
        }
    }
    return 0;
}

转载于:https://www.cnblogs.com/skylee03/p/6884996.html