POJ 3468 【线段树 成段更新 求区间和】.cpp

题意：

　　给出一些值 然后给出你一些命令 按照命令输出

　　命令：

　　C a b c 表示更新a ~ b的值加上 c

　　Q a b 表示查询a~b的和

输入：

　　n m 表示有n个数 m 个操作

　　接下来 m 行为命令行

 

思路：

　　就是简单的线段树成段更新求和　　

 

Tips：

　　※ 询问的时候要 pushdown

　　※ 注意范围问题吖~

 

Code：

　　

#include <stdio.h>
#include <cstring>
using namespace std;
#define LL long long

const int MAXN = 100010;
LL sum[MAXN<<2], lazy[MAXN<<2];

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

void pushdown(int rt, int x)
{
    if(lazy[rt] != 0) {
        lazy[rt<<1] += lazy[rt];
        lazy[rt<<1|1] += lazy[rt];
        sum[rt<<1] += (LL)(x-(x>>1))*lazy[rt];
        sum[rt<<1|1] += (LL)(x>>1)*lazy[rt];
        lazy[rt] = 0;
    }
}

void creat(int l, int r, int rt)
{
    lazy[rt] = 0;
    if(l == r) {
        scanf("%lld", &sum[rt]);
        return;
    }
    int mid = (l+r)>>1;
    creat(l, mid, rt<<1);
    creat(mid+1, r, rt<<1|1);
    pushup(rt);
}

void modify(int l, int r, int x, int L, int R, int rt)
{
    if(l <= L && r >= R) {
        lazy[rt] += (LL)x;
        sum[rt] += (LL)x*(R-L+1);
        return;
    }
    pushdown(rt, R-L+1);
    int mid = (L+R)>>1;
    if(l <= mid) modify(l, r, x, L, mid, rt<<1);
    if(r > mid) modify(l, r, x, mid+1, R, rt<<1|1);
    pushup(rt);
}

LL query(int l, int r, int L, int R, int rt)
{
    if(l <= L && r >= R) {
        return (LL)sum[rt];
    }
    pushdown(rt, R-L+1);///!!!!
    int mid = (L+R)>>1;
    LL ans = 0;
    if(l <= mid) ans += query(l, r, L, mid, rt<<1);
    if(r > mid) ans += query(l, r, mid+1, R, rt<<1|1);
    return ans;
}

int main()
{
    int i, j, k;
    int n, m;
    int a, b, c;
    char opt[2];
    while(scanf("%d %d", &n, &m) != EOF) {
        creat(1, n, 1);
        while(m--) {
            scanf("%s %d %d", opt, &a, &b);
            if(opt[0] == 'Q') {
                printf("%lld\n", query(a, b, 1, n, 1));
            } else if (opt[0] == 'C') {
                scanf("%d", &c);
                modify(a, b, c, 1, n, 1);
            }
        }
    }
    return 0;
}

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