线段树区间更新

今天按自己的思路写了写线段树的区间更新,就是POJ 3468,建树没有问题,然后再写更新函数,发现不对,调试了半天,原来是pushDown里面的变量打错了,晕死。。纠结了半天;

写完后,对线段树又有更深的了解了,数据结构真考代码能力,还有调试能力。深深认识到代码能力的不足,以后还是少套模板了。。

#include 
#define LL __int64

const int N = 500010;
LL num[N], sum[N], add[N], v;

void pushDown(int i, int l, int r) {
    if(add[i] != 0) {
        int mid = l + r >> 1;
        add[i << 1] += add[i];
        sum[i << 1] += (mid - l + 1) * add[i];  //[l, mid]代表左儿子区间
        add[i << 1 | 1] += add[i];
        sum[i << 1 | 1] += (r - mid) * add[i];  //[mid + 1, r]代表右儿子区间
        add[i] = 0;
    }
}

void update(int i, int l, int r, int ql, int qr, int val) {
    if(l > qr || ql > r)
        return ;
    if(l >= ql && r <= qr){
        sum[i] += (r - l + 1) * val;
        add[i] += val;
        return ;
    }
    pushDown(i, l, r);
    int mid = l + r >> 1;
    update(i << 1, l, mid, ql, qr, val);
    update(i << 1 | 1, mid + 1, r, ql, qr, val);
    sum[i] = sum[i << 1] + sum[i << 1 | 1];
}

LL query(int i, int l, int r, int ql, int qr) {
    if(l > qr || ql > r)
        return 0;
    if(l >= ql && r <= qr)
        return sum[i];
    pushDown(i, l, r);
    int mid = l + r >> 1;
    return query(i << 1, l, mid, ql, qr) 
        + query(i << 1 | 1, mid + 1, r, ql, qr);
}

int main() {
    int n, q, l, r;
    char ord;
    scanf("%d%d", &n, &q);
    for(int i = 1; i <= n; i++) {
        scanf("%lld", &num[i]);
        update(1, 1, n, i, i, num[i]);
    }
    while(q--) {
        scanf("\n%c", &ord);
        if(ord == 'Q') {
            scanf("%d%d", &l, &r);
            printf("%lld\n", query(1, 1, n, l, r));
        } else {
            scanf("%d%d%lld", &l, &r, &v);
            update(1, 1, n, l, r, v);
        }
    }
    return 0;
}


你可能感兴趣的:(ACM,数据结构)