T1082 线段树练习3 codevs

http://codevs.cn/problem/1082/

题目描述 Description

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

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

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

输入描述 Input Description

第一行一个正整数n，接下来n行n个整数，

再接下来一个正整数Q，每行表示操作的个数，

如果第一个数是1，后接3个正整数，

表示在区间[a,b]内每个数增加X,如果是2，

表示操作2询问区间[a,b]的和是多少。

pascal选手请不要使用readln读入

输出描述 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 <algorithm>
#include <iostream>
#define maxn 2000000+15
#define LL long long

using namespace std;

LL n,m,l,a,b,x;
struct node
{
    LL l,r,mid,dis,flag;
}tree[maxn*4];

void tree_up(LL now)
{
    tree[now].dis=tree[now*2].dis+tree[now*2+1].dis;
    return ;
}

void tree_build(LL now,LL l,LL r)
{
    tree[now].l=l,tree[now].r=r;
    if(l==r)
    {
        cin>>tree[now].dis;
        return ;
    }
    tree[now].mid=(l+r)/2;
    tree_build(now*2,l,tree[now].mid);
    tree_build(now*2+1,tree[now].mid+1,r);
    tree_up(now);
}

void tree_down(LL now)
{
    tree[now*2].flag+=tree[now].flag;
    tree[now*2].dis+=tree[now].flag*(tree[now*2].r-tree[now*2].l+1);
    tree[now*2+1].flag+=tree[now].flag;
    tree[now*2+1].dis+=tree[now].flag*(tree[now*2+1].r-tree[now*2+1].l+1);
    tree[now].flag=0;
    return ;
}

void tree_change_many(LL now,LL l,LL r,LL x)
{
    if(tree[now].l==l&&tree[now].r==r)
    {
        tree[now].flag+=x;
        tree[now].dis+=x*(tree[now].r-tree[now].l+1);
        return ;
    }
    if(tree[now].flag)    tree_down(now);
    if(tree[now].mid>=r)    tree_change_many(now*2,l,r,x);
    else if(tree[now].mid<l)    tree_change_many(now*2+1,l,r,x);
    else
    {
        tree_change_many(now*2,l,tree[now].mid,x);
        tree_change_many(now*2+1,tree[now].mid+1,r,x);
    }
    tree_up(now);
}

LL tree_query(LL now,LL l,LL r)
{
    if(tree[now].l==l&&tree[now].r==r)    return tree[now].dis;
    if(tree[now].flag)    tree_down(now);
    if(tree[now].mid>=r)    return tree_query(now*2,l,r);
    else if(tree[now].mid<l)    return tree_query(now*2+1,l,r);
    else    return tree_query(now*2,l,tree[now].mid)+tree_query(now*2+1,tree[now].mid+1,r);
}

int main()
{
    cin>>n;
    tree_build(1,1,n);
    cin>>m;
    while(m--)
    {
        cin>>x;
        if(x==1)
        {
            cin>>a>>b>>x;
            tree_change_many(1,a,b,x);
        }
        else
        {
            cin>>a>>b;
            cout<<tree_query(1,a,b)<<endl;
        }
    }
    return 0;
}