poj3468 A Simple Problem with Integers 线段树区间更新，区间求和

传送门：poj3468 A Simple Problem with Integers

题目大意

输入：第一行输入两个数N和Q。N代表有多少个元素，Q代表进行多少次操作
第二行数输入n个数 表示对应的初始值
然后下面的Q行代表操作：
C a b c把a到b区间的数增加c
Q a b输出a到b区间的数量之和
输出：每当有Q操作输出a到b区间的数量之和

解题思路

如果总是WA的话，可以测试这个数据：

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

ans

4
-82
-104
-147
-122
-100
-37
27
-73
7
14
21
25
-28

如果错的话，考虑一下是不是标记的事情，这道题目标记是可以重叠的。
你测试一下这个数据就知道了

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

AC代码

#include

const int MAXN = 200005;
#define lson left,mid,rt<<1
#define rson mid+1,right,rt<<1|1
typedef long long LL;  //lld
long long ans;

struct Node{
    int left;
    int right;
    long long  value;
    long long addMark;
    int mid()
    {
        return (left+right)>>1;
    }
}tree[MAXN*4];

void pushUp(int rt)
{
    tree[rt].value = tree[rt<<1].value + tree[rt<<1|1].value;
}
void pushDown(int rt)
{
    if(tree[rt].addMark !=0)
    {
        int m = (tree[rt].right - tree[rt].left+1);
        tree[rt<<1].value += ((m-m/2)*(tree[rt].addMark/m));
        tree[rt<<1|1].value += ((m/2)*(tree[rt].addMark/m));
        //在这道题目当中这里是+=不是=
        tree[rt<<1].addMark += ((m-m/2)*(tree[rt].addMark/m));
        tree[rt<<1|1].addMark += ((m/2)*(tree[rt].addMark/m));
        tree[rt].addMark = 0;
    }
}

void buildTree(int left,int right,int rt)
{
    tree[rt].left = left;
    tree[rt].right= right;
    tree[rt].addMark = 0;
    if(left == right)
    {
        scanf("%lld",&tree[rt].value);
        return ;
    }
    int mid = tree[rt].mid();
    buildTree(lson);
    buildTree(rson);
    pushUp(rt);
}


void updateTree(int left,int right,int rt,int L,int R,int val)
{
    pushDown(rt);
    if(left>=L && right<=R)
    {
        tree[rt].addMark +=(tree[rt].right - tree[rt].left+1)*val;
        tree[rt].value += (tree[rt].right - tree[rt].left+1)*val;
        return ;
    }
    int mid = tree[rt].mid();
    if(L<=mid)
        updateTree(lson,L,R,val);
    if(R>mid)
        updateTree(rson,L,R,val);
    pushUp(rt);
}

void queryTree(int left,int right,int rt,int L,int R)
{
    pushDown(rt);
    if(left>=L && right<=R)
    {
        ans += tree[rt].value;
        //printf("[%d]\n",ans);
        return ;
    }
    int mid = tree[rt].mid();
    if(L<=mid)
        queryTree(lson,L,R);
    if(R>mid)
        queryTree(rson,L,R);
}

int main()
{
    int n,q;
    char ch;
    int a,b,c;
    while(scanf("%d%d",&n,&q)!=EOF)
    {
        buildTree(1,n,1);
        while(q--)
        {
            getchar();
            scanf("%c",&ch);
            if(ch == 'C')
            {
                scanf("%d%d%d",&a,&b,&c);
                updateTree(1,n,1,a,b,c);
            }
            else if(ch == 'Q')
            {
                scanf("%d%d",&a,&b);
                ans = 0;
                queryTree(1,n,1,a,b);
                printf("%lld\n",ans);
            }
        }
    }
    return 0;
}