线段树 [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<<2];
ll sq[N<<2];
ll sum[N<<2];

void pushup(int rt)
{
    sum[rt]=sum[rt<<1]+sum[rt<<1|1];
    sq[rt]=sq[rt<<1]+sq[rt<<1|1];
}
void pushdown(int l,int r,int rt)
{
    if(add[rt])
    {
        int m=(l+r)>>1;
        add[rt<<1]+=add[rt];
        add[rt<<1|1]+=add[rt];

        sq[rt<<1]+=(2*add[rt]*sum[rt<<1]+(m-l+1)*add[rt]*add[rt]);
        sq[rt<<1|1]+=(2*add[rt]*sum[rt<<1|1]+(r-m)*add[rt]*add[rt]);

        sum[rt<<1]+=(m-l+1)*add[rt];
        sum[rt<<1|1]+=(r-m)*add[rt];
        add[rt]=0;
    }
}
void build(int l,int r,int rt)
{
    add[rt]=0;
    if(l==r)
    {
        scanf("%d",&sum[rt]);
        sq[rt]=sum[rt]*sum[rt];
        return;
    }
    int m=(l+r)>>1;
    build(l,m,rt<<1);
    build(m+1,r,rt<<1|1);
    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]+=2*sum[rt]*c+(r-l+1)*c*c;
        sum[rt]+=(r-l+1)*c;
        return;
    }
    int m=(l+r)>>1;
    pushdown(l,r,rt);
    if(R<=m) update(l,m,rt<<1,L,R,c);
    else if(L>m) update(m+1,r,rt<<1|1,L,R,c);
    else
    {
        update(l,m,rt<<1,L,m,c);
        update(m+1,r,rt<<1|1,m+1,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)>>1;
    pushdown(l,r,rt);
    if(R<=m) return query(l,m,rt<<1,L,R);
    else if(L>m) return query(m+1,r,rt<<1|1,L,R);
    else return query(l,m,rt<<1,L,m)+query(m+1,r,rt<<1|1,m+1,R);
}
int main()
{
    int n,m;
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        build(1,n,1);
        while(m--)
        {
            int op,a,b,c;
            scanf("%d",&op);
            if(op==1)
            {
                scanf("%d%d%d",&a,&b,&c);
                update(1,n,1,a,b,c);
            }
            else
            {
                scanf("%d%d",&a,&b);
                printf("%lld\n",query(1,n,1,a,b));
            }
        }
    }
    return 0;
}