poj 3468 A Simple Problem with Integers

线段树是一棵二叉树，树中的每一个结点表示了一个区间[a,b]。a,b通常是整数。每一个叶子节点表示了一个单位区间。对于每一个非叶结点所表示的结点[a,b]，其左儿子表示的区间为[a,(a+b)/2]，右儿子表示的区间为[(a+b)/2+1,b](除法去尾取整）。

该题每个节点保存一个区间下的总和以及可向下传递的累计量，若只保存总和，每次在更新节点和的时候，必然会更新其子节点一直到叶子节点，若保存一个可向下传递的量，则不需要每次更新到叶子节点，而是在query时才向下更新

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;


#define i64 long long


const int maxNode = 1000000;


struct Node
{
    int L;
    int R;
    i64 sum;
    i64 acc;
};


i64 sum = 0;


Node node[maxNode];


void buildTree(int root,int L,int R)
{
    node[root].L = L;
    node[root].R = R;
    node[root].sum = node[root].acc = 0;
    if(L==R)
        return;
    buildTree(root<<1,L,(L+R)>>1);
    buildTree((root<<1)+1,((L+R)>>1)+1,R);
}


void add(int root,int a,int b,int c)
{
    if(node[root].L==a&&node[root].R==b)
    {
        node[root].acc+=c;
        return;
    }
    node[root].sum += (b-a+1)*c;
    int mid = (node[root].L+node[root].R)>>1;
    if(b<=mid)
    {
        add(root<<1,a,b,c);
    }
    else if(a>mid)
    {
        add((root<<1)+1,a,b,c);
    }
    else
    {
        add(root<<1,a,mid,c);
        add((root<<1)+1,mid+1,b,c);
    }
}


void query(int root,int a,int b)
{
    if(node[root].L == a&&node[root].R==b)
    {
        sum += (node[root].sum+node[root].acc*(b-a+1));
        return;
    }
    node[root].sum+=node[root].acc*(node[root].R-node[root].L+1);
    int mid = (node[root].L+node[root].R)>>1;
    if(node[root].acc!=0)
    {
        add(root<<1,node[root].L,mid,node[root].acc);
        add((root<<1)+1,mid+1,node[root].R,node[root].acc);
    }
    node[root].acc = 0;
    if(b<=mid)
    {
        query(root<<1,a,b);
    }
    else if(a>mid)
    {
        query((root<<1)+1,a,b);
    }
    else
    {
        query(root<<1,a,mid);
        query((root<<1)+1,mid+1,b);
    }
}


int main()
{
    int n,q,num,a,b;
    char type;
    while(scanf("%d%d",&n,&q)!=EOF)
    {
        buildTree(1,1,n);
        for(int i = 1;i<=n;i++)
        {
            scanf("%d",&num);
            add(1,i,i,num);
        }
        for(int i = 1;i<=q;i++)
        {
            getchar();
            scanf("%c",&type);
            if(type=='Q')
            {
                scanf("%d %d",&a,&b);
                sum = 0;
                query(1,a,b);
                printf("%lld\n",sum);
            }
            else if(type == 'C')
            {
                scanf("%d %d %d",&a,&b,&num);
                add(1,a,b,num);
            }
        }
    }
    return 0;
}