A Simple Problem with Integers POJ - 3468（线段树）

最新推荐文章于 2022-08-28 01:34:07 发布

原创最新推荐文章于 2022-08-28 01:34:07 发布 · 118 阅读

0 ·

CC 4.0 BY-SA版权

数据结构/线段树专栏收录该内容

27 篇文章

订阅专栏

本文详细介绍了一种基于懒惰标记的区间更新查询算法实现，通过递归划分和标记传递，高效地处理区间内的元素增减操作。文章提供了完整的代码示例，包括建立树、查询、更新和下推等关键步骤。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

区间增减懒惰标记

#include <iostream>

using namespace std;
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
#define ll long long 
#define MAXN 111111

ll stree[MAXN<<2];
ll cont[MAXN<<2];


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

void build(int l,int r,int rt)
{
    cont[rt]=0;
    if(l==r)
    {
        scanf("%lld",&stree[rt]);
        return;
    }
    int mid=(l+r)>>1;
    build(lson);
    build(rson);
    pushup(rt);
}

void pushdown(int l,int r,int rt)
{
    if(cont[rt])
    {
        cont[rt<<1]+=cont[rt];
        cont[rt<<1|1]+=cont[rt];
        int mid=(l+r)>>1;
        stree[rt<<1]=stree[rt<<1]+(mid-l+1)*cont[rt];
        stree[rt<<1|1]=stree[rt<<1|1]+(r-(mid+1)+1)*cont[rt];
        cont[rt]=0;
    }
}

ll query(int L,int R,int l,int r,int rt)
{
    if(L<=l&&r<=R)
    {
        return stree[rt];
    }
    int mid=(l+r)>>1;
    pushdown(l,r,rt);
    ll ans=0;
    if(L<=mid) ans+=query(L,R,lson);
    if(R>mid) ans+=query(L,R,rson);
    return ans;
}

void update(int L,int R,int c,int l,int r,int rt)
{
    if(L<=l&&r<=R)
    {
        cont[rt]+=c;
        stree[rt]=stree[rt]+c*(r-l+1);
        return;
    }
    int mid=(l+r)>>1;
    pushdown(l,r,rt);
    if(L<=mid) update(L,R,c,lson);
    if(R>mid) update(L,R,c,rson);
    pushup(rt);
}

int main()
{
    int n,q;
    scanf("%d%d",&n,&q);
    build(1,n,1);
    char str[10];
    for(int i=0;i<q;i++)
    {
        int a,b,c;
        scanf("%s%d%d",str,&a,&b);
        if(str[0]=='Q')
            printf("%lld\n",query(a,b,1,n,1));
        else
        {
            scanf("%d",&c);
            update(a,b,c,1,n,1);
        }
    }
    return 0;
}