php求区间数的平方和,线段树，区间求和，求平方和，同乘同加

qn姐姐最好了~

qn姐姐给你了一个长度为n的序列还有m次操作让你玩，

1 l r 询问区间[l,r]内的元素和

2 l r 询问区间[l,r]内的元素的平方和

3 l r x 将区间[l,r]内的每一个元素都乘上x

4 l r x 将区间[l,r]内的每一个元素都加上x

输入描述:第一行两个数n,m

接下来一行n个数表示初始序列

就下来m行每行第一个数为操作方法opt，

若opt=1或者opt=2，则之后跟着两个数为l,r

若opt=3或者opt=4，则之后跟着三个数为l,r,x

操作意思为题目描述里说的

输出描述:对于每一个操作1,2，输出一行表示答案

示例1

输入

5 6

1 2 3 4 5

1 1 5

2 1 5

3 1 2 1

4 1 3 2

1 1 4

2 2 3

输出

15

55

16

41

备注:对于100%的数据 n=10000,m=200000 (注意是等于号)

保证所有询问的答案在long long 范围内

sum1[i] 表示区间和，sum2[i]表示区间平方和

add[i]表示同加延迟标记，mul[i]表示同乘延迟标记

注意更新延迟标记的时候，mul会对add产生影响#include

using namespace std;

#define lson l,mid,rt<<1

#define rson mid+1,r,rt<<1|1

typedef long long ll;

const int mx = 10005;

int n,m;

ll add[mx<<2],mul[mx<<2],sum1[mx<<2],sum2[mx<<2];

ll a[mx];

void pushup(int rt){

sum1[rt] = sum1[rt<<1] + sum1[rt<<1|1];

sum2[rt] = sum2[rt<<1] + sum2[rt<<1|1];

}

void build(int l,int r,int rt){

sum1[rt] = sum2[rt] = 0;

add[rt] = 0; mul[rt] = 1;

if(l == r){

sum1[rt] = a[l];

sum2[rt] = a[l]*a[l];

return ;

}

int mid = (l+r)>>1;

build(lson);

build(rson);

pushup(rt);

}

void pushdown(int rt,int len){

add[rt<<1] = add[rt] + add[rt<<1]*mul[rt];

add[rt<<1|1] = add[rt] + add[rt<<1|1]*mul[rt];

sum2[rt<<1] += 2*add[rt]*sum1[rt<<1] + (len-(len>>1))*add[rt]*add[rt];

sum2[rt<<1|1] += 2*add[rt]*sum1[rt<<1|1] + (len>>1)*add[rt]*add[rt];

sum1[rt<<1] = add[rt]*(len-(len>>1)) + sum1[rt<<1]*mul[rt];

sum1[rt<<1|1] = add[rt]*(len>>1) + sum1[rt<<1|1]*mul[rt];

mul[rt<<1] *= mul[rt];

mul[rt<<1|1] *= mul[rt];

mul[rt] = 1;

add[rt] = 0;

}

void update(int l,int r,int rt,int L,int R,int val,int type){

if(l >= L && r <= R){

if(type == 3){

mul[rt] *= val;

add[rt] *= val;

sum1[rt] *= val;

sum2[rt] *= (val*val);

return ;

}else{

add[rt] += val;

sum2[rt] += (2*val*sum1[rt] + val*val*(r-l+1));

sum1[rt] += (val*(r-l+1));

return ;

}

}

pushdown(rt,r-l+1);

int mid = (l+r)>>1;

if(L <= mid) update(lson,L,R,val,type);

if(R > mid) update(rson,L,R,val,type);

pushup(rt);

}

ll q1,q2;

ll res;

void query(int l,int r,int rt,int L,int R){

if(l >= L && r <= R){

q1 += sum1[rt];

q2 += sum2[rt];

return ;

}

int mid = (l+r)>>1;

res = 0;

pushdown(rt,r-l+1);

if(L <= mid) query(lson,L,R);

if(R > mid) query(rson,L,R);

//pushup(rt);

}

int main(){

while(~scanf("%d%d",&n,&m)){

for(int i = 1; i <= n; i++){

scanf("%lld",&a[i]);

}

build(1,n,1);

int op,l,r;

ll v;

while(m--){

scanf("%d",&op);

q1 = q2 = res = 0;

if(op == 1){

scanf("%d%d",&l,&r);

query(1,n,1,l,r);

printf("%lld\n",q1);

}

else if(op == 2){

scanf("%d%d",&l,&r);

query(1,n,1,l,r);

printf("%lld\n",q2);

}

else if(op == 3){

scanf("%d%d%lld",&l,&r,&v);

update(1,n,1,l,r,v,3);

}

else if(op == 4){

scanf("%d%d%lld",&l,&r,&v);

update(1,n,1,l,r,v,4);

}

}

}

}