POJ 3468 A Simple Problem with Integers (线段树成段更新) 解题报告

一道线段树成段更新的入门题。所谓成段更新（也叫延迟操作、懒惰标记），就是说在线段树的数组外，再开一个同样大小的lazy数组。当更新区间[l,r]中的每个元素时，不用更新到每个叶子节点，只需要更新到[l,r]所在的区间，同时在lazy数组中做记录，表示该节点的子节点尚未更新。

       这样一来，当查询时，如果查询的区间完全包含[l,r]，则不用更新子节点，直接用这个节点的数据；如果查询的区间在[l,r]区间的子节点，那么要先根据lazy数组更新该节点的子节点，然后再去查询，每次更新只更新一层子节点，不用一下更新到底，用大白话讲，用到哪儿，更新到哪儿。

       有个细节要注意一下：在根据lazy数组更新子节点（即push_down的过程）时，有可能子节点的lazy标记中还有数据，表示子节点的子节点尚未更新，所以应该是lazy[rt<<1]+=lazy[rt]，而不是lazy[rt<<1]=lazy[rt]。

       另外，这道题要注意lazy数组也要用long long类型，因为在乘法操作时会超int范围。

       时间复杂度为O(mnlogn)，实际测试结果如下，两份一样的代码，用C++交比G++快一些（注意，这个规律不具有普遍性，有时候G++比C++快一些）：

C++代码：

#include <stdio.h>
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
typedef long long lint;
const int MAXN = 100005;
lint seg_tree[MAXN*4];
lint lazy[MAXN*4];           
//lazy标记表示本节点数据已经更新，但是子节点没有更新，注意，本题中lazy数组也必须用long long，在乘法运算时会超int的
void push_up(int rt)
{
    seg_tree[rt] = seg_tree[rt<<1] + seg_tree[rt<<1|1];
}
void push_down(int l,int r,int rt)
{
    int m = (l+r)>>1;
    seg_tree[rt<<1] += lazy[rt]*(m-l+1);
    seg_tree[rt<<1|1] += lazy[rt]*(r-m);
    //注意：这里的延迟标记是+=，不是=，因为子节点上本身可能还有没给子节点的子节点更新的数据，直接覆盖会出错的
    lazy[rt<<1] += lazy[rt];       
    lazy[rt<<1|1] += lazy[rt];      
    lazy[rt] = 0;
}
void build(int l,int r,int rt)
{
    if (l == r)
        scanf("%lld",&seg_tree[rt]);
    else
    {
        int m = (l+r) >> 1;
        build(lson);
        build(rson);
        push_up(rt);
    }
}
lint query(int L,int R,int l,int r,int rt)
{
    if (L == l && R == r)
        return seg_tree[rt];
    if (lazy[rt] != 0 && l != r)
        push_down(l,r,rt);
    int m = (l+r)>>1;
    if (R <= m)
        return query(L,R,lson);
    else if (L > m)
        return query(L,R,rson);
    else
        return query(L,m,lson)+query(m+1,R,rson);
}
void update(int L,int R,lint num,int l,int r,int rt)
{
    if (lazy[rt] != 0 && l != r)
        push_down(l,r,rt);
    if (L == l && R == r)
    {
        seg_tree[rt] += (r-l+1) * num;
        lazy[rt] = num;
    }
    else
    {
        int m = (l+r)>>1;
        if (R <= m)
            update(L,R,num,lson);
        else if (L > m)
            update(L,R,num,rson);
        else
        {
            update(L,m,num,lson);
            update(m+1,R,num,rson);
        }
        push_up(rt);
    }
}
int main()
{
    int n,q,i,a,b;
    lint c;
    char ch;
    scanf("%d %d",&n,&q);
    build(1,n,1);
    for (i = 0; i < q; ++i)
    {
        getchar();
        scanf("%c",&ch);
        if (ch == 'Q')
        {
            scanf("%d %d",&a,&b);
            printf("%lld\n",query(a,b,1,n,1));
        }
        else
        {
            scanf("%d %d %lld",&a,&b,&c);
            update(a,b,c,1,n,1);
        }
    }
    return 0;
}