BZOJ 1798 AHOI2009 Seq 维护序列 线段树
题目大意:维护一个序列,提供三种操作:
1.将区间中每个点的权值乘上一个数
2.将区间中每个点的权值加上一个数
3.求一段区间的和对p取模的值
2631的超^n级弱化版。写2631之前能够拿这个练练手。。。
线段树区间改动,让学校的大神指导了一下ZKW的区间改动方法,非常好理解,可是代码还是快不了。
。
。
回头再改改代码吧 可能是我写的太渣了
#include<cstdio> #include<cstring> #include<iostream> #include<algorithm> #define M 100100 using namespace std; typedef long long ll; struct abcd{ ll sum,siz,add_mark,times_mark; void add(ll x); void times(ll x); }tree[263000]; int m,n,p,q; void abcd :: add(ll x) { sum+=x*siz; sum%=p; add_mark+=x; add_mark%=p; } void abcd :: times(ll x) { sum*=x; sum%=p; add_mark*=x; add_mark%=p; times_mark*=x; times_mark%=p; } void Build_Tree() { int i; for(i=q+1;i<=q+n;i++) { scanf("%d",&tree[i].sum); tree[i].siz=1; tree[i].add_mark=0; tree[i].times_mark=1; } for(i=q-1;i;i--) { tree[i].sum=tree[i<<1].sum+tree[i<<1|1].sum; tree[i].sum%=p; tree[i].siz=tree[i<<1].siz+tree[i<<1|1].siz; tree[i].add_mark=0; tree[i].times_mark=1; } } int stack[M],top; void Push_Down(int x) { if(x>=q) return ; if(tree[x].times_mark^1) { tree[x<<1 ].times(tree[x].times_mark); tree[x<<1|1].times(tree[x].times_mark); tree[x].times_mark=1; } if(tree[x].add_mark) { tree[x<<1 ].add(tree[x].add_mark); tree[x<<1|1].add(tree[x].add_mark); tree[x].add_mark=0; } } void Push_Down(int x,int y) { for(x+=q-1,y+=q+1;x^y;x>>=1,y>>=1) stack[++top]=x,stack[++top]=y; for(;x;x>>=1) stack[++top]=x; while(top) Push_Down(stack[top--]); } void Add(int x,int y,ll z) { for(x+=q-1,y+=q+1;x^y^1;x>>=1,y>>=1) { if(~x&1)tree[x^1].add(z); if( y&1)tree[y^1].add(z); } } void Times(int x,int y,ll z) { for(x+=q-1,y+=q+1;x^y^1;x>>=1,y>>=1) { if(~x&1)tree[x^1].times(z); if( y&1)tree[y^1].times(z); } } void Push_Up(int x) { if(x>=q) return ; tree[x].sum=tree[x<<1].sum+tree[x<<1|1].sum; tree[x].sum%=p; } void Push_Up(int x,int y) { for(x+=q-1,y+=q+1;x^y;x>>=1,y>>=1) Push_Up(x),Push_Up(y); for(;x;x>>=1) Push_Up(x); } ll Get_Ans(int x,int y) { int re=0; for(x+=q-1,y+=q+1;x^y^1;x>>=1,y>>=1) { if(~x&1)re+=tree[x^1].sum,re%=p; if( y&1)re+=tree[y^1].sum,re%=p; } return re; } int main() { int i,x,y,c; ll z; cin>>n>>p; for(q=1;q<=n+1;q<<=1); Build_Tree(); cin>>m; for(i=1;i<=m;i++) { scanf("%d",&c); if(c==1) { scanf("%d%d%lld",&x,&y,&z); Push_Down(x,y); Times(x,y,z); Push_Up(x,y); } else if(c==2) { scanf("%d%d%lld",&x,&y,&z); Push_Down(x,y); Add(x,y,z); Push_Up(x,y); } else { scanf("%d%d",&x,&y); Push_Down(x,y); printf("%lld\n", Get_Ans(x,y) ); } } }