【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=0,f=1;
    char ch=getchar();
    while(ch<'0'||ch>'9')
    {
        if(ch=='-')f=-1;
        ch=getchar();
    }
    while(ch>='0'&&ch<='9')
    {
        x=x*10+ch-'0';
        ch=getchar();
    }
    return x*f;
}
//**************************************************************************************
struct ss
{
    int l,r;
    ll sum;
    int tag;
}tr[4*200005];
int n;
int a[200005];
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)>>1;
    build(k<<1,l,mid);
    build(k<<1|1,mid+1,r);
    tr[k].sum=tr[k<<1].sum+tr[k<<1|1].sum;
}
void pushdown(int k)
{
    int x=tr[k].r-tr[k].l+1;
    tr[k<<1].tag+=tr[k].tag;
    tr[k<<1|1].tag+=tr[k].tag;
    tr[k<<1].sum+=(x-(x>>1))*tr[k].tag;
    tr[k<<1|1].sum+=(x>>1)*tr[k].tag;
    tr[k].tag=0;
}
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+1)*x;
        return ;
    }
    if(tr[k].tag)pushdown(k);
    int mid=(tr[k].l+tr[k].r)>>1;
    if(b<=mid)update(k<<1,a,b,x);
    else if(a>mid)update(k<<1|1,a,b,x);
    else {
       update(k<<1,a,mid,x);
   update(k<<1|1,mid+1,b,x);
    }
    tr[k].sum=tr[k<<1].sum+tr[k<<1|1].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)>>1;
    if(b<=mid)return ask(k<<1,a,b);
    else if(a>mid) return ask(k<<1|1,a,b);
    else {
        return (ask(k<<1,a,mid)+ask(k<<1|1,mid+1,b));
    }
}
int main()
{

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