[OpenJudge] 平方和

平方和

总时间限制: 3000ms 内存限制: 65536kB

描述

给出n(1<=n<=500000)个数字，下标从1开始

执行m(1<=m<=500000)次操作，操作可分为两种：

1 l r x:将区间[l,r]内的每个数加上x  1<=l<=r<=n -100<=x<=100

2 l r:输出区间[l,r]内每个数平方之和

输入

多组输入
第一行两个整数n m
第二行n个整数
余下m行表示m个操作意义叙述于上

输出

对应每个2操作输出相应的值

样例输入

5 3
1 1 1 1 1
2 1 5
1 1 5 1
2 1 5

样例输出

5
20
 
线段树成段更新

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
#define N 500005
#define ll long long
 
ll add[N<<];
ll sq[N<<];
ll sum[N<<];
 
void pushup(int rt)
{
    sum[rt]=sum[rt<<]+sum[rt<<|];
    sq[rt]=sq[rt<<]+sq[rt<<|];
}
void pushdown(int l,int r,int rt)
{
    if(add[rt])
    {
        int m=(l+r)>>;
        add[rt<<]+=add[rt];
        add[rt<<|]+=add[rt];
 
        sq[rt<<]+=(*add[rt]*sum[rt<<]+(m-l+)*add[rt]*add[rt]);
        sq[rt<<|]+=(*add[rt]*sum[rt<<|]+(r-m)*add[rt]*add[rt]);
 
        sum[rt<<]+=(m-l+)*add[rt];
        sum[rt<<|]+=(r-m)*add[rt];
        add[rt]=;
    }
}
void build(int l,int r,int rt)
{
    add[rt]=;
    if(l==r)
    {
        scanf("%d",&sum[rt]);
        sq[rt]=sum[rt]*sum[rt];
        return;
    }
    int m=(l+r)>>;
    build(l,m,rt<<);
    build(m+,r,rt<<|);
    pushup(rt);
}
void update(int l,int r,int rt,int L,int R,int c)
{
    if(l==L && r==R)
    {
        add[rt]+=c;
        sq[rt]+=*sum[rt]*c+(r-l+)*c*c;
        sum[rt]+=(r-l+)*c;
        return;
    }
    int m=(l+r)>>;
    pushdown(l,r,rt);
    if(R<=m) update(l,m,rt<<,L,R,c);
    else if(L>m) update(m+,r,rt<<|,L,R,c);
    else
    {
        update(l,m,rt<<,L,m,c);
        update(m+,r,rt<<|,m+,R,c);
    }
    pushup(rt);
}
ll query(int l,int r,int rt,int L,int R)
{
    if(l==L && r==R)
    {
        return sq[rt];
    }
    int m=(l+r)>>;
    pushdown(l,r,rt);
    if(R<=m) return query(l,m,rt<<,L,R);
    else if(L>m) return query(m+,r,rt<<|,L,R);
    else return query(l,m,rt<<,L,m)+query(m+,r,rt<<|,m+,R);
}
int main()
{
    int n,m;
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        build(,n,);
        while(m--)
        {
            int op,a,b,c;
            scanf("%d",&op);
            if(op==)
            {
                scanf("%d%d%d",&a,&b,&c);
                update(,n,,a,b,c);
            }
            else
            {
                scanf("%d%d",&a,&b);
                printf("%lld\n",query(,n,,a,b));
            }
        }
    }
    return ;
}