线段树区间更新

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

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

#include <cstdio>
#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;
}


### 线段树区间修改的实现方法 线段树是一种高效的数据结构,用于处理动态区间查询和修改操作。对于区间修改的操作,通常会引入懒惰传播(Lazy Propagation)机制来优化性能。 #### 基本概念 在支持区间修改的情况下,线段树通过维护一个额外的`lazy[]`数组记录尚未传递给子节点的延迟更新信息。这种设计可以减少不必要的递归调用次数,从而提高效率[^1]。 #### 关键函数说明 以下是实现线段树区间修改的核心部分: 1. **构建线段树** 构建过程与普通的线段树相同,初始化时需确保`lazy[]`数组全部置零。 2. **Push Down 函数** 当访问某个节点并发现该节点存在未解决的延迟标记时,需要将其影响向下传递至子节点,并清除当前节点上的标记。 ```cpp void pushDown(int k, int l, int r) { if (lazy[k]) { // 如果有延迟标记 int mid = (l + r) / 2; add(k * 2, l, mid, lazy[k]); // 更新左孩子 add(k * 2 + 1, mid + 1, r, lazy[k]); // 更新右孩子 lazy[k] = 0; // 清除当前节点的延迟标记 } } ``` 3. **Add 函数** `add()`负责执行具体的区间修改逻辑。如果目标区间完全覆盖当前节点,则直接应用修改;否则继续分解到子节点上。 ```cpp void add(int k, int l, int r, int ql, int qr, int val) { if (ql <= l && r <= qr) { // 完全覆盖的情况 s[k] += (r - l + 1) * val; // 修改当前区间的总和 lazy[k] += val; // 设置延迟标记 return; } int mid = (l + r) / 2; pushDown(k, l, r); // 下推延迟标记 if (ql <= mid) add(k * 2, l, mid, ql, qr, val); if (qr > mid) add(k * 2 + 1, mid + 1, r, ql, qr, val); s[k] = s[k * 2] + s[k * 2 + 1]; // 合并左右孩子的结果 } ``` 4. **Query 函数** 查询过程中也需要注意是否存在延迟标记,若有则先进行下推操作再继续查找。 ```cpp long long query(int k, int l, int r, int ql, int qr) { if (ql <= l && r <= qr) { // 完全覆盖的情况 return s[k]; } int mid = (l + r) / 2; pushDown(k, l, r); // 下推延迟标记 long long res = 0; if (ql <= mid) res += query(k * 2, l, mid, ql, qr); if (qr > mid) res += query(k * 2 + 1, mid + 1, r, ql, qr); return res; } ``` 以上即为完整的线段树区间修改实现方案[^2][^3]。 ### 时间复杂度分析 每次修改或查询操作最多涉及从根节点到叶子节点的一条路径上的所有节点,因此时间复杂度均为\( O(\log N) \)[^4]。
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值