wikioi 1082 線段樹練習 3

http://wikioi.com/problem/1082/

線段樹，區間更新，區間求和，最好用到懶散標記。

線段樹的學習，可以參考

http://www.notonlysuccess.com/index.php/segment-tree-complete/

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<string.h>
#include<stdlib.h>
#include<algorithm>
using namespace std;
# define inf -10000000
long long sum[800100],tag[800100];
void pushup(int root)//向上更新sum
{
    sum[root] = sum[2*root] + sum[2*root + 1];
}
void pushdown(int root,int num)//向下修改，更新左右孩子
{
    tag[2*root] += tag[root];
    tag[2*root + 1] += tag[root];
    sum[2*root] +=(num - num/2) * tag[root];
    sum[2*root + 1] += (num/2) * tag[root];
    tag[root] = 0;//tag  標記爲0
}
void build(int l,int r,int root)
{
    tag[root] = 0;//tag  初始化
    if(l == r)
    {
        scanf("%lld",&sum[root]);
        return ;
    }
    else
    {
        int mid = (l+r)/2;
        build(l,mid,2*root);//建立左子樹
        build(mid+1,r,2*root+1);//建立右子樹
        pushup(root);
    }
}
void update(int L,int R,int v,int l,int r,int root)
{
    if(L <= l && R >= r)//找到區間，更新sum  並修改 懶散標記
    {
        sum[root] += (r - l + 1)*v; // 區間內每個值都增加
        tag[root] += v;
        return ;
    }
    pushdown(root,r-l+1);
    int mid = (l+r)/2;
    if(L <= mid)update(L,R,v,l,mid,2*root);
    if(R>mid)update(L,R,v,mid+1,r,2*root+1);
    pushup(root);// 向上傳遞sum
}
long long getsum(int L,int R,int l,int r,int root)
{
    if(L<= l && R >= r) return sum[root];//找到查詢區間，返回sum 值
    pushdown(root,r-l+1);// 向下邊修改邊查詢，查詢過程中訪問到的區間會用到增量
    int mid = (l + r)/2;
    long long res = 0;
    if(L <= mid) res += getsum(L,R,l,mid,2*root);
    if(R>mid) res += getsum(L,R,mid+1,r,2*root+1);
    return res;
}

int main()
{
    int n,m,a,b,c,d;
    scanf("%d",&n);
    build(1,n,1);
    scanf("%d",&m);
    while(m--)
    {
        scanf("%d%d%d",&a,&b,&c);
        if(a == 1)
        {
            scanf("%d",&d);
            update(b,c,d,1,n,1);
        }
        else
        {
            printf("%lld\n",getsum(b,c,1,n,1));
        }
    }
    return 0;
}