暑期训练狂刷系列——poj 3468 A Simple Problem with Integers （线段树+区间更新）

题目连接：

　　http://poj.org/problem?id=3468

题目大意：

　　给出n个数，有两种操作：

　　　　1："C a b c"，[a,b]中的每一个数都加上c。

　　　　2："Q a b"，求[a,b]中每个数相加的和。

解题思路:

　　线段树更新到每一个节点的话，由于节点数目和查询次数原因会tle，所以在每一个节点内定义一个标志变量表示当前节点的下一层为更新，每次查询时候有需要的话在更新到下一层。

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
const int maxn = 400010;
const int INF = 0x3f3f3f3f;
#define LL __int64
struct node
{
    LL L, R;
    LL sum, add;
    LL Mid()
    {
        return (L + R) / 2;
    }
};
node tree[maxn];
LL res;

void build(LL root, LL l, LL r)
{
    tree[root].L = l;
    tree[root].R = r;
    tree[root].sum = tree[root].add = 0;

    if (l == r)
        return ;
    build (2*root+1, l, tree[root].Mid());
    build (2*root+2, tree[root].Mid()+1, r);
}
void insert (LL root, LL s, LL e, LL x)
{
    tree[root].sum += x * (e - s + 1);
    if (tree[root].L == s && e == tree[root].R)//更新到区间
        {
            tree[root].add += x;
            return ;
        }
    if (e <= tree[root].Mid())
        insert (2*root+1, s, e, x);
    else if (tree[root].Mid() < s)
        insert (2*root+2, s, e, x);
    else
    {
        insert (2*root+1, s, tree[root].Mid(), x);
        insert (2*root+2, tree[root].Mid()+1, e, x);
    }
}
void query (LL root, LL s, LL e)
{

    if (tree[root].L == s && e == tree[root].R)
    {
        res += tree[root].sum;
        return ;
    }
    if (tree[root].add)
    {//向下继续更新
        tree[2*root+1].add += tree[root].add;
        tree[2*root+2].add += tree[root].add;
        tree[2*root+1].sum += tree[root].add * (tree[2*root+1].R - tree[2*root+1].L + 1);
        tree[2*root+2].sum += tree[root].add * (tree[2*root+2].R - tree[2*root+2].L + 1);
        tree[root].add = 0;
    }
    if (e <= tree[root].Mid())
        query (2*root+1, s, e);
    else if (tree[root].Mid() < s)
        query (2*root+2, s, e);
    else
    {
        query (2*root+1, s, tree[root].Mid());
        query (2*root+2, tree[root].Mid()+1, e);
    }
}
int main ()
{
    LL n, m, num;
    while (scanf ("%I64d %I64d", &n, &m) != EOF)
    {
        build (0, 1, n);
        for (int i=1; i<=n; i++)
        {
            scanf ("%I64d", &num);
            insert (0, i, i, num);
        }
        char str[2];
        LL s, e;
        while (m --)
        {
            scanf ("%s %I64d %I64d", str, &s, &e);
            if (str[0] == 'Q')
            {
                res = 0;
                query (0, s, e);
                printf ("%I64d
", res);
            }
            else
            {
                scanf ("%I64d", &num);
                insert (0, s, e, num);
            }
        }
    }
    return 0;
}