codevs 1082 线段树练习 3 --分块练习

#include <ctype.h>
#include <cstdio>
#include <cmath>
#define N 500
typedef long long LL;
void read(LL &x)
{
    x=0;bool f=0;char ch=getchar();
    while(!isdigit(ch))
    {
        if(ch==‘-‘) f=1;
        ch=getchar();
    }
    while(isdigit(ch))
    {
        x=x*10+ch-‘0‘;
        ch=getchar();
    }
    x=f?(~x)+1:x;
}
LL C,m,belong[N*N],cnt,tag[N],sum[N],beg[N],en[N],n,a[N*N]; 
LL min(LL a,LL b) {return a>b?b:a;}
LL max(LL a,LL b) {return a>b?a:b;} 
void update(LL x,LL y,LL z)
{
    for(LL i=belong[x];i<=belong[y];i++)
    {
        if(x<=beg[i]&&y>=en[i]) tag[i]+=z,sum[i]+=(en[i]-beg[i]+1)*z;
        else for(LL j=max(beg[i],x);j<=min(en[i],y);j++) a[j]+=z,sum[i]+=z;
    }
}
LL query(LL x,LL y)
{
    LL ans=0;
    for(LL i=belong[x];i<=belong[y];i++)
    {
        if(x<=beg[i]&&y>=en[i]) ans+=sum[i];
        else for(LL j=max(beg[i],x);j<=min(en[i],y);j++) ans+=a[j]+tag[i];
    }
    return ans;
}
int main()
{
    read(n);
    for(LL i=1;i<=n;i++) read(a[i]);
    C=sqrt(n);
    for(LL i=1;i<=n;i+=C)
    {
        beg[++cnt]=i;
        en[cnt]=min(i+C-1,n);
    }
    for(LL i=1;i<=cnt;i++)
    {
        for(LL j=beg[i];j<=en[i];j++) belong[j]=i,sum[i]+=a[j];
    }
    read(m);
    for(LL type,x,y,z;m--;)
    {
        read(type);
        read(x);
        read(y);
        if(type==1)
        {
            read(z);
            update(x,y,z);
        }
        else printf("%lld\n",query(x,y));
    }
    return 0;
}