【codevs】1082 线段树练习 3 <区间修改+区间和>

题目连接   http://codevs.cn/problem/1082/

Description

给你N个数，有两种操作：

1：给区间[a,b]的所有数增加X

2：询问区间[a,b]的数的和。

Input

第一行一个正整数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 ouput

9

数据范围及提示 Data Size & Hint

数据范围

1<=n<=200000

1<=q<=200000

代码

#include <cstdio>
#include <cmath>
#include <cstring>
#include <ctime>
#include <iostream>
#include <algorithm>
#include <set>
#include <vector>
#include <queue>
#include <typeinfo>
#include <map>
#include <stack>
typedef long long ll;
#define inf 0x7fffffff
using namespace std;
inline ll read()
{
    ll x=,f=;
    char ch=getchar();
    while(ch<''||ch>'')
    {
        if(ch=='-')f=-;
        ch=getchar();
    }
    while(ch>=''&&ch<='')
    {
        x=x*+ch-'';
        ch=getchar();
    }
    return x*f;
}
//**************************************************************************************
struct ss
{
    int l,r;
    ll sum;
    int tag;
}tr[*];
int n;
int a[];
void build(int k,int l,int r)
{
    tr[k].l=l;
    tr[k].r=r;
    if(l==r) {tr[k].sum=a[l];return;}
    int mid=(l+r)>>;
    build(k<<,l,mid);
    build(k<<|,mid+,r);
    tr[k].sum=tr[k<<].sum+tr[k<<|].sum;
}
void pushdown(int k)
{
    int x=tr[k].r-tr[k].l+;
    tr[k<<].tag+=tr[k].tag;
    tr[k<<|].tag+=tr[k].tag;
    tr[k<<].sum+=(x-(x>>))*tr[k].tag;
    tr[k<<|].sum+=(x>>)*tr[k].tag;
    tr[k].tag=;
}
void update(int k,int a,int b,int x)
{
    if(a==tr[k].l&&b==tr[k].r)
    {
        tr[k].tag+=x;
        tr[k].sum+=(b-a+)*x;
        return ;
    }
    if(tr[k].tag)pushdown(k);
    int mid=(tr[k].l+tr[k].r)>>;
    if(b<=mid)update(k<<,a,b,x);
    else if(a>mid)update(k<<|,a,b,x);
    else {
       update(k<<,a,mid,x);
   update(k<<|,mid+,b,x);
    }
    tr[k].sum=tr[k<<].sum+tr[k<<|].sum;
}
ll ask(int k,int a,int b)
{
    if(a==tr[k].l&&b==tr[k].r){return tr[k].sum;}
    if(tr[k].tag) pushdown(k);
    int mid=(tr[k].l+tr[k].r)>>;
    if(b<=mid)return ask(k<<,a,b);
    else if(a>mid) return ask(k<<|,a,b);
    else {
        return (ask(k<<,a,mid)+ask(k<<|,mid+,b));
    }
}
int main()
{

    scanf("%d",&n);
    for(int i=;i<=n;i++)
    {
        scanf("%d",&a[i]);
    }
    build(,,n);
    int q;
   scanf("%d",&q);
    for(int i=;i<=q;i++)
    {
        int x,aa,b,mm;
        scanf("%d",&mm);
        if(mm==)
        {
            scanf("%d%d%d",&aa,&b,&x);
            update(,aa,b,x);
        }
        else
        {
            scanf("%d%d",&aa,&b);
            printf("%lld\n",ask(,aa,b));
        }
    }
    return ;
}