POJ 3468 A Simple Problem with Integers（线段树）

题目链接

题意 ： 给你n个数，进行两种操作，第一种是将a到b上所有元素都加上c，第二种是查询a到b上所有元素之和输出。

思路 ： 线段树，以前写过博客，但是现在在重刷，风格改变，，所以重新写一篇。。。。

#include <cstdio>
#include <cstring>
#include <iostream>
#define LL long long
using namespace std ;

LL lz[100010*4],p[100100*4] ;

void pushup(int rt)
{
    p[rt] = p[rt << 1] + p[rt << 1 | 1] ;
}
void pushdown(int rt,int m)
{
    if(lz[rt])
    {
        lz[rt << 1] += lz[rt] ;
        lz[rt << 1 | 1] += lz[rt] ;
        p[rt << 1] += (m-m/2)*lz[rt] ;
        p[rt << 1 | 1] += m/2 * lz[rt] ;
        lz[rt] = 0 ;
    }
}
void build(int l,int r,int rt)
{
    lz[rt] = 0 ;
    if(l == r)
    {
        scanf("%I64d%*c",&p[rt]) ;
        return ;
    }
    int mid = (l+r) >> 1 ;
    build(l,mid,rt << 1) ;
    build(mid+1,r,rt << 1 | 1) ;
    pushup(rt) ;
}
LL query(int L,int R,int l,int r,int rt)
{
    LL sum = 0 ;
    if(l >= L && R >= r)
    {
        return p[rt] ;
    }
    pushdown(rt,r-l+1) ;
    int mid = (l+r) >> 1;
    if(mid >= L)
        sum += query(L,R,l,mid,rt << 1) ;
    if(mid < R)
        sum += query(L,R,mid+1,r,rt << 1 | 1) ;
    return sum ;
}
void update(int L,int R, int l,int r,int sc,int rt)
{
    if(l >= L && r <= R)
    {
        lz[rt] += sc ;
        p[rt] += (r-l+1)*sc ;
        return  ;
    }
    pushdown(rt,r-l+1) ;
    int mid = (l+r) >> 1 ;
    if(mid >= L)
        update(L,R,l,mid,sc,rt << 1) ;
    if(mid < R)
        update(L,R,mid+1,r,sc,rt << 1 | 1) ;
    pushup(rt) ;
}
int main()
{
    int N,Q ,a,b;
    LL c ;
    char ch[3] ;
    while(~scanf("%d %d",&N,&Q))
    {
        build(1,N,1) ;
        for(int i = 0 ; i < Q ; i++)
        {
            scanf("%s",ch) ;
            if(ch[0] == 'Q')
            {
                scanf("%d %d",&a,&b) ;
                LL sum = query(a,b,1,N,1) ;
                printf("%I64d
",sum) ;
            }
            else
            {
                scanf("%d %d %I64d",&a,&b,&c) ;
                update(a,b,1,N,c,1) ;
            }
        }
    }
    return 0 ;
}