C - A Simple Problem with Integers ——线段树_区间更新

这篇博客介绍了如何使用线段树解决区间更新的问题,重点讨论了懒惰标记(lazy propagation)在优化区间更新效率上的作用。作者通过对比单点更新导致超时的代码与应用懒惰标记后AC的代码,阐述了线段树在区间查询和更新中的应用策略。
摘要由CSDN通过智能技术生成

Think:
1知识点:线段树-区间更新-区间查询-lazy标记
2题意分析:区间更新+区间查询
3反思:
1>单点更新会超时
2>lazy标记区间更新AC

vjudge题目链接

以下为Time Limit Exceeded代码——单点更新整个区间超时

#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;

typedef long long LL;
const int N = 101400;

struct Node{
    LL num;
}node[N<<2];

void Build(int l, int r, int rt);
void Updata(int rt);
LL find(int L, int R, int l, int r, int rt);
void add_v(int L, int R, int v, int l, int r, int rt);

int main(){
    int n, m, L, R, v;
    char st[14];
    while(~scanf("%d %d", &n, &m)){
        Build(1, n, 1);
        while(m--){
            scanf("%s", st);
            if(st[0] == 'Q'){
                scanf("%d %d", &L, &R);
                printf("%lld\n", find(L, R, 1, n, 1));
            }
            else if(st[0] == 'C'){
                scanf("%d %d %d", &L, &R, &v);
                add_v(L, R, v, 1, n, 1);
            }
        }
    }
}
void Build(int l, int r, int rt){
    if(l == r){
        scanf("%lld", &node[rt].num);
        return;
    }
    int mid = (l+r)/2;
    Build(l, mid, rt<<1);
    Build(mid+1, r, rt<<1|1);
    Updata(rt);
}
void Updata(int rt){
    node[rt].num = node[rt<<1].num + node[rt<<1|1].num;
}
LL find(int L, int R, int l, int r, int rt){
    if(L <= l && r <= R)
        return node[rt].num;
    int mid = (l+r)/2;
    LL sum = 0;
    if(L <= mid)
        sum += find(L, R, l, mid, rt<<1);
    if(R > mid)
        sum += find(L, R, mid+1, r, rt<<1|1);
    return sum;
}
void add_v(int L, int R, int v, int l, int r, int rt){
    if(l == r){
        node[rt].num += v;
        return;
    }
    int mid = (l+r)/2;
    if(L <= mid)
        add_v(L, R, v, l, mid, rt<<1);
    if(R > mid)
        add_v(L, R, v, mid+1, r, rt<<1|1);
    Updata(rt);
}

以下为Accepted代码——lazy标记区间更新

#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;

typedef long long LL;
const int N = 101400;

struct Node{
    LL num;
}node[N<<2];
LL lazy[N<<2];/*lazy标记,用则更新*/

void Build(int l, int r, int rt);
void Updata(int rt);
void down(int rt, int len);/*更新lazy数组标记的当前根节点的下一层*/
LL find(int L, int R, int l, int r, int rt);
void add_v(int L, int R, int v, int l, int r, int rt);

int main(){
    int n, m, L, R, v;
    char st[14];
    while(~scanf("%d %d", &n, &m)){
        Build(1, n, 1);
        while(m--){
            scanf("%s", st);
            if(st[0] == 'Q'){
                scanf("%d %d", &L, &R);
                printf("%lld\n", find(L, R, 1, n, 1));
            }
            else if(st[0] == 'C'){
                scanf("%d %d %d", &L, &R, &v);
                add_v(L, R, v, 1, n, 1);
            }
        }
    }
}
void Build(int l, int r, int rt){
    lazy[rt] = 0;
    if(l == r){
        scanf("%lld", &node[rt].num);
        return;
    }
    int mid = (l+r)/2;
    Build(l, mid, rt<<1);
    Build(mid+1, r, rt<<1|1);
    Updata(rt);
}
void Updata(int rt){
    node[rt].num = node[rt<<1].num + node[rt<<1|1].num;
}
void down(int rt, int len){
    if(lazy[rt]){
        lazy[rt<<1] += lazy[rt];
        lazy[rt<<1|1] += lazy[rt];
        node[rt<<1].num += lazy[rt]*(len - (len>>1));/*lazy[rt]*(l-mid+1)*/
        node[rt<<1|1].num += lazy[rt]*(len>>1);/*lazy[rt]*(l-(mid+1)+1)**/
        lazy[rt] = 0;
    }
}
LL find(int L, int R, int l, int r, int rt){
    if(L <= l && r <= R)
        return node[rt].num;
    down(rt, r-l+1);
    int mid = (l+r)/2;
    LL sum = 0;
    if(L <= mid)
        sum += find(L, R, l, mid, rt<<1);
    if(R > mid)
        sum += find(L, R, mid+1, r, rt<<1|1);
    return sum;
}
void add_v(int L, int R, int v, int l, int r, int rt){
    if(L <= l && r <= R){
        lazy[rt] += v;
        node[rt].num += v*(r-l+1);
        return;
    }
    down(rt, r-l+1);
    int mid = (l+r)/2;
    if(L <= mid)
        add_v(L, R, v, l, mid, rt<<1);
    if(R > mid)
        add_v(L, R, v, mid+1, r, rt<<1|1);
    Updata(rt);
}
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值