Codevs 1082 线段树练习 3

1082 线段树练习 3
时间限制: 3 s
空间限制: 128000 KB
题目等级 : 大师 Maste
传送门
题目描述 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
数据范围及提示 Data Size & Hint
数据范围
1<=n<=200000
1<=q<=200000

/*
裸线段树.
区间修改+区间查询.
*/
#include<iostream>
#include<cstdio>
#define MAXN 200001
#define ll long long
using namespace std;
ll n,m,tot,cut,aa[MAXN+10];
struct data
{
    int l,r;
    int lc,rc;
    ll sum;
    ll bj;
}tree[MAXN*4];
void build(int l,int r)
{
    int k=++cut;
    tree[k].l=l;
    tree[k].r=r;
    if(l==r)
    {
        tree[k].sum=aa[l];
        return ;
    }
    int mid=(l+r)>>1;
    tree[k].lc=cut+1;
    build(l,mid);
    tree[k].rc=cut+1;
    build(mid+1,r);
    tree[k].sum=tree[tree[k].lc].sum+tree[tree[k].rc].sum;
}
void updata(int k)
{
    tree[tree[k].lc].sum+=
    tree[k].bj*(tree[tree[k].lc].r-tree[tree[k].lc].l+1);
    tree[tree[k].rc].sum+=
    tree[k].bj*(tree[tree[k].rc].r-tree[tree[k].rc].l+1);
    tree[tree[k].lc].bj+=tree[k].bj;
    tree[tree[k].rc].bj+=tree[k].bj;
    tree[k].bj=0;
}
void add(int k,int l,int r,int add1)
{
    if(l<=tree[k].l&&tree[k].r<=r)
    {
        tree[k].bj+=add1;
        tree[k].sum+=add1*(tree[k].r-tree[k].l+1);
        return ;
    }
    if(tree[k].bj) updata(k);
    int mid=(tree[k].l+tree[k].r)>>1;
    if(l<=mid) add(tree[k].lc,l,r,add1);
    if(r>mid) add(tree[k].rc,l,r,add1);
    tree[k].sum=tree[tree[k].lc].sum+tree[tree[k].rc].sum;
}
ll query(int k,int l,int r)
{
    if(l<=tree[k].l&&tree[k].r<=r) return tree[k].sum;
    ll tot=0;
    if(tree[k].bj) updata(k);
    int mid=(tree[k].l+tree[k].r)>>1;
    if(l<=mid) tot+=query(tree[k].lc,l,r);
    if(r>mid) tot+=query(tree[k].rc,l,r);
    tree[k].sum=tree[tree[k].lc].sum+tree[tree[k].rc].sum;
    return tot;
}
int main()
{
    int x,a,b,add1;
    cin>>n;
    for(int i=1;i<=n;i++)
    {
        cin>>aa[i];
    }
    build(1,n);
    cin>>m;
    for(int i=1;i<=m;i++)
    {
        scanf("%d",&x);
        if(x==1)
        {
            scanf("%d %d %d",&a,&b,&add1);
            add(1,a,b,add1);
        }
        else
        {
            scanf("%d %d",&a,&b);
            printf("%lld
",query(1,a,b));
        }
    }
    return 0;   
}