线段树 - A Simple Problem with Integers - POJ - 3468

线段树 - A Simple Problem with Integers - POJ - 3468

题意：

$给定长度为N的序列A_1,A_2,...,A_N，共有两种操作：$

$1、“Clrd”，表示把A[l],A[l+1],\dots ,A[r]都加上d。$

$2、“Qlr”，表示询问数列中区间[l,r]的数的和。$

$共Q组询问。$

Sample Input

10 5
1 2 3 4 5 6 7 8 9 10
Q 4 4
Q 1 10
Q 2 4
C 3 6 3
Q 2 4

Sample Output

4
55
9
15

数据范围：

$1\le N,Q\le 100000，-1000000000\le A_i\le 1000000000，-10000\le d\le 10000.$
$Timelimit:5000ms,Casetimelimit:2000ms,Memorylimit:131072kB$

---

分析：

$两种操作分别是区间修改和区间求和，可以通过差分数组，转化为单点修改，再通过树状数组求前缀和。$
$详见——$《树状数组 - 一个简单的整数问题》。

$这里用线段树来解决这个问题。$

$做区间加法，需要在节点结构体中添加懒标记add。$

$表示某段区间内的所有数都增加add。$

$需要注意的是，懒标记只有在查询或者修改子区间的时候，才会向下传递。每次传递后，要将父节点的懒标记清空。$

$因为我们要保证每个节点表示的区间内的懒标记要相同，不能整个区间的几段add的值不同。$

$对懒标记有个pushdown操作，记父节点为root，向下传递懒标记add:$

$左区间left:left.add+=root.add，left.sum+=cn{t}_{left}\times left.add，其中cn{t}_{left}为左区间中数的个数。$

$右区间同理。最后不要忘记清空root.add=0。$

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>

#define ll long long

using namespace std;

const int N=1e5+10;

struct node
{
    int l,r;
    ll sum,add;
}tr[N*4];

int n,m;
int w[N];

void pushup(int u)
{
    tr[u].sum=tr[u<<1].sum+tr[u<<1|1].sum;
}

void pushdown(int u)
{
    node &root=tr[u], &left=tr[u<<1], &right=tr[u<<1|1];
    if(root.add)
    {
        left.add+=root.add, left.sum+=(ll)(left.r-left.l+1)*root.add;
        right.add+=root.add, right.sum+=(ll)(right.r-right.l+1)*root.add;
        root.add=0;
    }
}

void build(int u,int l,int r)
{
    if(l==r) tr[u]={l,r,w[r],0};
    else
    {
        tr[u]={l,r};
        int mid=l+r>>1;
        build(u<<1,l,mid),build(u<<1|1,mid+1,r);
        pushup(u);
    }
}

void modify(int u,int l,int r,int d)
{
    if(l<=tr[u].l&&tr[u].r<=r)
    {
        tr[u].sum+=(ll)(tr[u].r-tr[u].l+1)*d;
        tr[u].add+=d;
    }
    else  //区间分裂开了，有几段的懒标记不同
    {
        pushdown(u);
        int mid=tr[u].l+tr[u].r>>1;
        if(l<=mid) modify(u<<1,l,r,d);
        if(r>mid) modify(u<<1|1,l,r,d);
        pushup(u);
    }
}

ll query(int u,int l,int r)
{
    if(l<=tr[u].l&&tr[u].r<=r) return tr[u].sum;

    pushdown(u);
    int mid=tr[u].l+tr[u].r>>1;
    ll sum=0;
    if(l<=mid) sum+=query(u<<1,l,r);
    if(r>mid) sum+=query(u<<1|1,l,r);
    return sum;
}

int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++) scanf("%d",&w[i]);
    build(1,1,n);
    
    char op[2];
    int l,r,d;
    while(m--)
    {
        scanf("%s%d%d",op,&l,&r);
        if(*op=='Q') printf("%lld\n",query(1,l,r));
        else
        {
            scanf("%d",&d);
            modify(1,l,r,d);
        }
    }
    
    return 0;
}