poj3468(线段树区间更新&区间求和模板)

题目链接: http://poj.org/problem?id=3468

题意: 输入 n, m表初始有 n 个数, 接下来 m 行输入, Q x y 表示询问区间 [x, y]的和；

C x y z 表示区间 [x, y] 内所有数加上 z ；

思路: 线段树区间更新&区间求和模板；

代码:

 #include <iostream>
 #include <stdio.h>
 #define ll long long
 #define lson l, mid, rt << 1
 #define rson mid + 1, r, rt << 1 | 1
 using namespace std;

 const int MAXN = 1e5 + ;
 ll sum[MAXN << ];
 ll add[MAXN << ];

 void push_up(int rt){//向上更新
     sum[rt] = sum[rt << ] + sum[rt <<  | ];
 }

 void push_down(int rt, int m){
     if(add[rt]){//若有标记,则将标记向下移动一层
         add[rt << ] += add[rt];
         add[rt <<  | ] += add[rt];
         sum[rt << ] += (m - (m >> )) * add[rt];
         sum[rt <<  | ] += (m >> ) * add[rt];
         add[rt] = ;//取消本层标记
     }
 }

 void build(int l, int r, int rt){//建树
     add[rt] = ;
     if(l == r){
         scanf("%lld", &sum[rt]);
         return;
     }
     int mid = (l + r) >> ;
     build(lson);
     build(rson);
     push_up(rt);//向上更新
 }

 void update(int L, int R, ll key, int l, int r, int rt){//区间更新
     if(L <= l && R >= r){
         sum[rt] += (r - l + ) * key;
         add[rt] += key;
         return;
     }
     push_down(rt, r - l + );//向下更新
     int mid = (l + r) >> ;
     if(L <= mid) update(L, R, key, lson);
     if(R > mid) update(L, R, key, rson);
     push_up(rt);//向上更新
 }

 ll query(int L, int R, int l, int r, int rt){//区间求和
     if(L <= l && R >= r) return sum[rt];
     push_down(rt, r - l + );//向下更新
     int mid = (l + r) >> ;
     ll ans = ;
     if(L <= mid) ans += query(L, R, lson);
     if(R > mid) ans += query(L, R, rson);
     return ans;
 }

 int main(void){
     int n, m;
     scanf("%d%d", &n, &m);
     build(, n, );
     while(m--){
         char str[];
         int x, y;
         ll z;
         scanf("%s", str);
         if(str[] == 'C'){
             scanf("%d%d%lld", &x, &y, &z);
             update(x, y, z, , n, );
         }else{
             scanf("%d%d", &x, &y);
             printf("%lld\n", query(x, y, , n, ));
         }
     }
 }