Codevs 4919 线段树练习4 (线段树)

2019-04-14 16:14发布

class="markdown_views prism-atom-one-light">

题意分析

emmmm WA了一天
我一直以为是区间修改写挂了,然后发现其实策略PUSHUP,PUSHDOWN写的不好。 首先考虑,在建树的时候,叶子节点不必保存数值。
其次,在更新的时候,只需要更新更新数值模7的结果。 如果模7为0,则不必更新。
在更新的时候,只需要对保存模7余几的那个数组进行相应修改,别的不用管了。 PUSHUP:将节点左右儿子的结果进行合并。
PUSHDOWN:对于叶子节点不必PUSHDOWN,对于lazy = 0的节点也不必PUSHDOWN。在PUSHDOWN的时候标记下传,同时对左右儿子的结果进行更新。

代码总览

#include #include #include #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;iscanf(" %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 ",Query(a,b,1)); } } return 0; }