【线段树-区间更新求区间和】CDOJ 1057

秋实大哥与花

Time Limit: 3000/1000MS (Java/Others)     Memory Limit: 65535/65535KB (Java/Others)

Submit  Status

秋实大哥是一个儒雅之人，昼听笙歌夜醉眠，若非月下即花前。

所以秋实大哥精心照料了很多花朵。现在所有的花朵排成了一行，每朵花有一个愉悦值。

秋实大哥每天要对着某一段连续的花朵歌唱，然后这些花朵的愉悦值都会增加一个相同的值 v ( v 可能为负)。

同时他想知道每次他唱完歌后这一段连续的花朵的愉悦值总和是多少。

Input

第一行有一个整数n，表示花朵的总数目。
第二行包含n个整数ai，表示第i朵花初始的愉悦值。
第三行包含一个整数m，表示秋实大哥唱了m天的歌。
接下来m行，每行包含三个整数l  r  v，表示秋实大哥对着[l,r]这个区间内的花朵歌唱，每朵花的愉悦值增加了v。
1≤n，m，ai，|v|≤100000，1≤l≤r≤n。

Output

输出共 m m行，第 i i行表示秋实大哥完成第 i i天的歌唱后，那一段花朵的愉悦值总和。

Sample input and output

Sample Input	Sample Output
3 0 0 0 3 1 2 1 1 2 -1 1 3 1	2 0 3

Source

题意：

思路：区间更新，求区间和；

开辟一个sum数组记录每个节点所对应区间的和；

再开辟一个数组作为lazy_tag的作用；那么什么是lazy_tag?  

1.建树的时候就会更新完成sum数组-- PushUp回溯更新sum数组 ；

2.比如C(2,5,2)，从节点1开始向下一直寻找到有完整区间的节点，然后加上lazy_tag；沿路遇到有lazy_tag的节点，会把lazy_tag給沿途Push_dow下来；

直到更新结束；(lazy_tag是个总称)

3.然后Q(1,4)，上一步的C操作会使得区间(2,2)(3,3)(4,5)有lazy_tag标记，而这次的Q(1,4)查询呢，会找到区间(1,3),(4,4);(4,4)又会经过(4,5)，会把(4,5)

的lazy_tag PushDown；

//一定要注意pushup的回溯！

代码：

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

typedef long long ll;
const int N=100010;

ll sum[N<<2]; //结点i所代表的区间的和；
ll add[N<<2]; //lazy_tag；

struct NODE
{
    int l,r;
    int mid()
    {
        return (l+r)>>1;
    }
} tree[N<<2];

void PushUp(int rt)             //更新sum数组
{
    sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}

void PushDown(int rt,int len)   //更新add数组与sum数组
{
    //给出父区间长度，可得两个子区间长度；
    if(add[rt])
    {
        add[rt<<1] += add[rt];
        add[rt<<1|1] += add[rt];
        sum[rt<<1] += add[rt]*(len-(len>>1)); //父区间[1,3]长度为3;则左子区间长度为3-3/2，右子区间长度为3/2；
        sum[rt<<1|1] += add[rt]*(len>>1);
        add[rt] = 0;                          //lazy_tag传下去了，自己就变成0；
    }
}

void BuildTree(int l,int r,int rt)
{
    tree[rt].l=l;
    tree[rt].r=r;
    add[rt]=0,sum[rt]=0;
    if(l==r)
    {
        scanf("%I64d",&sum[rt]);
        return ;
    }

    int m=tree[rt].mid();
    BuildTree(l,m,rt<<1);
    BuildTree(m+1,r,rt<<1|1);
    PushUp(rt);                           //回溯，得到结点rt所表示的区间和；
}

void UpdateTree(int l,int r,int c,int rt)
{
    if(tree[rt].l==l&&tree[rt].r==r)
    {
        add[rt] +=c;
        sum[rt] +=(ll)c*(r-l+1);
        return ;                          //区间完全重合掉，就return;
    }
    if(tree[rt].l==tree[rt].r) return ;

    PushDown(rt,tree[rt].r-tree[rt].l+1); //给出父区间长度，可得两个子区间长度；
    int m=tree[rt].mid();
    if(r<=m) UpdateTree(l,r,c,rt<<1);
    else if(l>m) UpdateTree(l,r,c,rt<<1|1);
    else
    {
        UpdateTree(l,m,c,rt<<1);
        UpdateTree(m+1,r,c,rt<<1|1);
    }
    PushUp(rt);  //回溯以得到更新后的区间值；
}

ll Query(int l,int r,int rt)
{
    if(tree[rt].l==l&&tree[rt].r==r)
    {
        return sum[rt];
    }
    PushDown(rt,tree[rt].r-tree[rt].l+1);
    int m=tree[rt].mid();

//    ll res=0;
    if(r<=m) return Query(l,r,rt<<1);
    else if(l>m) return Query (l,r,rt<<1|1);
    else return (Query(l,m,rt<<1)+Query(m+1,r,rt<<1|1));
}

int main()
{
    int n,m;
    while(scanf("%d",&n)!=EOF)
    {
        BuildTree(1,n,1);

        scanf("%d",&m);
        while(m--)
        {
            int a,b,c;
            scanf("%d%d%d",&a,&b,&c);
            UpdateTree(a,b,c,1);
            printf("%lld\n",Query(a,b,1));
        }
    }
    return 0;
}