线段树，区间求和，求平方和，同乘同加

链接：https://www.nowcoder.com/acm/contest/200/B
来源：牛客网
 

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<bits/stdc++.h>
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){
    mul[rt<<1] *= mul[rt];//
    mul[rt<<1|1] *= mul[rt];//
    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] = sum2[rt<<1]*mul[rt]*mul[rt]+ 2*add[rt]*sum1[rt<<1] + (len-(len>>1))*add[rt]*add[rt];
    sum2[rt<<1|1] = sum2[rt<<1|1]*mul[rt]*mul[rt]+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;
    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);
            }
        }
    }   
}