codevs1082线段树练习3（线段树）

1082 线段树练习 3
时间限制: 3 s
空间限制: 128000 KB
题目等级 : 大师 Master
题目描述 Description
给你N个数，有两种操作：
1：给区间[a,b]的所有数增加X
2：询问区间[a,b]的数的和。
输入描述 Input Description
第一行一个正整数n，接下来n行n个整数，
再接下来一个正整数Q，每行表示操作的个数，
如果第一个数是1，后接3个正整数，
表示在区间[a,b]内每个数增加X,如果是2，
表示操作2询问区间[a,b]的和是多少。
输出描述 Output Description
对于每个询问输出一行一个答案
样例输入 Sample Input
3
1
2
3
2
1 2 3 2
2 2 3
样例输出 Sample Output
9
数据范围
1<=n<=200000
1<=q<=200000
传递标记时node[s].value+=z*(r-l+1)。。
附上本蒟蒻的代码：

#include<cstdio>
using namespace std;
struct edge
{
    long long value;
};
edge node[800001];
int n;
long long ans,a[200001],delta[800001]={0};

int read()
{
    int w=0,c=1;
    char ch=getchar();
    while (ch<'0' || ch>'9')
      {
        if (ch=='-')
          c=-1;
        ch=getchar();
      }
    while (ch>='0' && ch<='9')
      {
        w=w*10+ch-'0';
        ch=getchar();
      }
    return w*c;
}

void update(int s)
{
    node[s].value=node[s*2].value+node[s*2+1].value;
}

void build(int s,int l,int r)
{
    if (l==r)
      {
        node[s].value+=a[l];
        return;
      }
    build(s*2,l,(l+r)/2);
    build(s*2+1,(l+r)/2+1,r);
    update(s);
}

void paint(int s,int z,int l,int r)
{
    node[s].value+=z*(r-l+1);
    delta[s]+=z;
}

void pushdown(int s,int l,int r)
{
    int mid;
    mid=(l+r)/2;
    paint(s*2,delta[s],l,mid);
    paint(s*2+1,delta[s],mid+1,r);
    delta[s]=0;
}

void insert(int s,int l,int r,int x,int y,int z)
{
    int mid;
    if (x<=l && y>=r)
      {
        paint(s,z,l,r);
        return;
      }
    pushdown(s,l,r);
    mid=(l+r)/2;
    if (x<=mid)
      insert(s*2,l,mid,x,y,z);
    if (y>mid)
      insert(s*2+1,mid+1,r,x,y,z);
    update(s);
}

void query(int s,int l,int r,int x,int y)
{
    int mid;
    if (x<=l && y>=r)
      {
        ans+=node[s].value;
        return;
      }
    if (delta[s]!=0)
      pushdown(s,l,r);
    mid=(l+r)/2;
    if (x<=mid)
      query(s*2,l,mid,x,y);
    if (y>mid)
      query(s*2+1,mid+1,r,x,y);
}

int main()
{
    int i,p,x,y,z,m;
    n=read();
    for (i=1;i<=n;i++)
      a[i]=read();
    build(1,1,n);
    m=read();
    for (i=1;i<=m;i++)
      {
        p=read();
        if (p==1)
          {
            x=read();
            y=read();
            z=read();
            insert(1,1,n,x,y,z);
          }
        if (p==2)
          {
            x=read();
            y=read();
            ans=0;
            query(1,1,n,x,y);
            printf("%lld\n",ans);
          }
      }
    return 0;
}