CQUOJ 24914 线段树写个爽（线段树TAT）

http://acm.cqu.edu.cn:8888/oj/problem_show.php?pid=24914

一个有四种操作的线段树，区间加和，区间加等差数列，区间赋值，区间查询。

真是写死人并且那个坑调了好久才调出来。

区间加等差数列可以维护个首项和公差就行了，一样pushdown，然后区间赋值的时候，需要把前两个的标记全部清除，在pushdown的时候也是，左右子树的前两个操作的标记要全部清除，而且要先执行第三个操作，因为如果前两个操作在这个区间，再来第三个，就会把他们清除，这无所谓，但是如果是第三个操作先在这个区间，然后后两个操作也在这个区间，pushdown的时候如果先执行了前两个，再执行第三个的话，左右子树的标记就会被第三个操作给清除，所以第三个操作应该优先执行，反正他的标记只能被他自己清除。这里错了好久。

代码：

#include <map>
#include <set>
#include <stack>
#include <queue>
#include <cmath>
#include <string>
#include <vector>
#include <cstdio>
#include <cctype>
#include <cstring>
#include <sstream>
#include <cstdlib>
#include <iostream>
#include <algorithm>
#pragma comment(linker, "/STACK:102400000,102400000")

using namespace std;
#define   MAX           100005
#define   MAXN          6005
#define   maxnode       15
#define   sigma_size    30
#define   lson          l,m,rt<<1
#define   rson          m+1,r,rt<<1|1
#define   lrt           rt<<1
#define   rrt           rt<<1|1
#define   middle        int m=(r+l)>>1
#define   LL            long long
#define   ull           unsigned long long
#define   mem(x,v)      memset(x,v,sizeof(x))
#define   lowbit(x)     (x&-x)
#define   pii           pair<int,int>
#define   bits(a)       __builtin_popcount(a)
#define   mk            make_pair
#define   limit         10000

//const int    prime = 999983;
const int    INF   = 0x3f3f3f3f;
const LL     INFF  = 0x3f3f;
const double pi    = acos(-1.0);
//const double inf   = 1e18;
const double eps   = 1e-8;
const LL     mod    = 1e9+7;
const ull    mx    = 133333331;

/*****************************************************/
inline void RI(int &x) {
char c;
while((c=getchar())<'0' || c>'9');
x=c-'0';
while((c=getchar())>='0' && c<='9') x=(x<<3)+(x<<1)+c-'0';
}
/*****************************************************/

LL sum[MAX<<2];
LL add[MAX<<2];
LL a1[MAX<<2],a2[MAX<<2];
LL col[MAX<<2];
LL aa[MAX<<2],bb[MAX<<2],cc[MAX<<2];
void pushup(int rt){
sum[rt]=sum[lrt]+sum[rrt];
}

void build(int l,int r,int rt){
add[rt]=a1[rt]=a2[rt]=col[rt]=0;
aa[rt]=bb[rt]=cc[rt]=0;
if(l==r){
scanf("%lld",&sum[rt]);
return ;
}
middle;
build(lson);
build(rson);
pushup(rt);
}

void pushdown(int rt,int m){
if(cc[rt]){
cc[lrt]=1;
aa[lrt]=0;add[lrt]=0;
bb[lrt]=0;a1[lrt]=0,a2[lrt]=0;
cc[rrt]=1;
aa[rrt]=0;add[rrt]=0;
bb[rrt]=0;a1[rrt]=0,a2[rrt]=0;
col[lrt]=col[rt];
col[rrt]=col[rt];
sum[lrt]=col[rt]*(m-(m>>1));
sum[rrt]=col[rt]*(m>>1);
col[rt]=0;
cc[rt]=0;
}
if(aa[rt]){
aa[lrt]=1;
aa[rrt]=1;
aa[rt]=0;
add[lrt]+=add[rt];
add[rrt]+=add[rt];
sum[lrt]+=add[rt]*(m-(m>>1));
sum[rrt]+=add[rt]*(m>>1);
add[rt]=0;
}
if(bb[rt]){
bb[lrt]=1;
bb[rrt]=1;
bb[rt]=0;
a1[lrt]+=a1[rt];
a1[rrt]+=a1[rt]+a2[rt]*(m-(m>>1));
a2[lrt]+=a2[rt];
a2[rrt]+=a2[rt];
sum[lrt]+=(2*a1[rt]+a2[rt]*(m-(m>>1)-1))*(m-(m>>1))/2;
sum[rrt]+=(2*a1[rt]+a2[rt]*(2*m-(m>>1)-1))*(m>>1)/2;
a1[rt]=0;a2[rt]=0;
}
}

void update1(int l,int r,int rt,int L,int R,int d){
if(L<=l&&r<=R){
sum[rt]+=(r-l+1)*d;
add[rt]+=d;
aa[rt]=1;
return;
}
middle;
pushdown(rt,r-l+1);
if(L<=m) update1(lson,L,R,d);
if(R>m) update1(rson,L,R,d);
pushup(rt);
}

void update2(int l,int r,int rt,int L,int R,int d){
if(L<=l&&r<=R){
sum[rt]+=(LL)(r+l-2*L+2)*(r-l+1)*d/2;
a1[rt]+=(LL)d*(l-L+1);
a2[rt]+=d;
bb[rt]=1;
return;
}
middle;
pushdown(rt,r-l+1);
if(L<=m) update2(lson,L,R,d);
if(R>m) update2(rson,L,R,d);
pushup(rt);
}

void update3(int l,int r,int rt,int L,int R,int d){
if(L<=l&&r<=R){
sum[rt]=(r-l+1)*d;
col[rt]=d;
cc[rt]=1;
aa[rt]=0;add[rt]=0;
bb[rt]=0;a1[rt]=0,a2[rt]=0;
return;
}
middle;
pushdown(rt,r-l+1);
if(L<=m) update3(lson,L,R,d);
if(R>m) update3(rson,L,R,d);
pushup(rt);
}

LL query(int l,int r,int rt,int L,int R){
if(L<=l&&r<=R) return sum[rt];
middle;
pushdown(rt,r-l+1);
LL ans=0;
if(L<=m) ans+=query(lson,L,R);
if(R>m) ans+=query(rson,L,R);
pushup(rt);
return ans;
}

int main(){
int t;
cin>>t;
while(t--){
int n,m;
cin>>n>>m;
build(1,n,1);
while(m--){
int op;
scanf("%d",&op);
if(op==1){
int a,b,c;
scanf("%d%d%d",&a,&b,&c);
update1(1,n,1,a,b,c);
}
else if(op==2){
int a,b,c;
scanf("%d%d%d",&a,&b,&c);
update2(1,n,1,a,b,c);
}
else if(op==3){
int a,b,c;
scanf("%d%d%d",&a,&b,&c);
update3(1,n,1,a,b,c);
}
else{
int a,b;
scanf("%d%d",&a,&b);
printf("%lld\n",query(1,n,1,a,b));
}
}
}
return 0;
}