Codevs 4919 线段树练习4 （线段树）

题意分析

emmmm WA了一天
我一直以为是区间修改写挂了，然后发现其实策略PUSHUP，PUSHDOWN写的不好。

首先考虑，在建树的时候，叶子节点不必保存数值。
其次，在更新的时候，只需要更新更新数值模7的结果。 如果模7为0，则不必更新。
在更新的时候，只需要对保存模7余几的那个数组进行相应修改，别的不用管了。

PUSHUP：将节点左右儿子的结果进行合并。
PUSHDOWN：对于叶子节点不必PUSHDOWN，对于lazy = 0的节点也不必PUSHDOWN。在PUSHDOWN的时候标记下传，同时对左右儿子的结果进行更新。

代码总览

#include <cstdio>
#include <cstring>
#include <algorithm>
#define nmax 100010
using namespace std;
struct Tree{
    int l,r;
    int lazy;
    int tans[7];
    int mid(){
        return (l+r)>>1;
    }
};
Tree tree[nmax<<2];
int  num[nmax];
int N,Q,a,b;
int c;
char op[40];
void change(int rt ,int val){
    int temp[7] = {0};
    for(int i = 0;i<7;++i) temp[ (i+ val) % 7] = tree[rt].tans[i];
    for(int i = 0;i<7;++i) tree[rt].tans[i] = temp[i];
}
void PushUp(int rt){
    for(int i = 0;i<7;++i) tree[rt].tans[i] = tree[rt<<1].tans[i] + tree[rt<<1|1].tans[i];
}
void PushDown(int rt){
    if(!tree[rt].lazy || tree[rt].l == tree[rt].r) return;
    tree[rt<<1].lazy += tree[rt].lazy;
    tree[rt<<1|1].lazy += tree[rt].lazy;
    change(rt<<1,tree[rt].lazy);
    change(rt<<1|1,tree[rt].lazy);
    tree[rt].lazy = 0;
}
void Build(int l, int r, int rt){
    tree[rt].l = l; tree[rt].r = r;
    tree[rt].lazy = 0;
    if(l == r){
        tree[rt].tans[num[l] % 7] = 1;
        return;
    }
    Build(l,tree[rt].mid(),rt<<1);
    Build(tree[rt].mid()+1,r,rt<<1|1);
    PushUp(rt);
}
void UpdateInterval(int val, int l, int r, int rt) {

    if(tree[rt].l >= l && tree[rt].r <= r){
        tree[rt].lazy += val;
        change(rt,val);
        return;
    }
    PushDown(rt);
    if(l <= tree[rt].mid()) UpdateInterval(val,l,r,rt<<1);
    if(r > tree[rt].mid()) UpdateInterval(val,l,r,rt<<1|1);
    PushUp(rt);
}


int Query(int l,int r,int rt){
    PushDown(rt);
    if(l <= tree[rt].l && tree[rt].r <= r ) return tree[rt].tans[0];
    int ans = 0;
    if(l<= tree[rt].mid()) ans += Query(l,r,rt<<1);
    if(r> tree[rt].mid()) ans += Query(l,r,rt<<1|1);
    return ans;
}
int main(){
    scanf("%d",&N);
    for(int i = 1;i<=N;++i) scanf("%d",&num[i]);
    Build(1,N,1);
    scanf("%d",&Q);
    for(int i = 0;i<Q;++i){
        scanf(" %s %d %d",op,&a,&b);
        if(!strcmp(op,"add")){
            scanf("%d",&c);
            c = c % 7;
            if(c) UpdateInterval(c,a,b,1);
        }else{
            printf("%d\n",Query(a,b,1));
        }
    }
    return 0;
}