A Simple Problem with Integers POJ - 3468-优快云博客

本文链接：https://blog.youkuaiyun.com/Haipai1998/article/details/74857709

本文介绍了一种使用线段树进行区间更新和查询的方法。通过递归将更新操作分解到子区间，实现区间更新的同时标记节点以提高效率。文章详细解释了查询和更新过程中的懒惰传播机制。

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

/*If I get TLE , it is good.If I get AC,it's NICE !*/
#include <stdio.h>
#include <iostream>
#include <algorithm>
#include <string.h>
using namespace std;
typedef long long ll;
const int MAXN=1e5+10;
ll a[MAXN];
ll ans=0;
struct node
{
    int l,r;
    ll sum;
    ll c;
}segtree[4*MAXN];

void query(int l,int r,int root)
{
    if(l==segtree[root].l && r==segtree[root].r)
    {
        ans+=segtree[root].sum;
        return ;
    }
    if(segtree[root].c) //查询的时候注意要向下传递信息
    {
        segtree[root*2].c+=segtree[root].c;
        segtree[root*2].sum+=segtree[root].c*(segtree[root*2].r-segtree[root*2].l+1);
        segtree[root*2+1].c+=segtree[root].c;
        segtree[root*2+1].sum+=segtree[root].c*(segtree[root*2+1].r-segtree[root*2+1].l+1);
        segtree[root].c=0;
    }
    int mid=(segtree[root].l+segtree[root].r)/2;
    if(r<=mid) query(l,r,root*2);
    else if(l>=mid+1)   query(l,r,root*2+1);
    else
    {
        query(l,mid,root*2);
        query(mid+1,r,root*2+1);
    }
}
// 主要想讲一下对区间更新过程中需要注意的。对于我下面的代码，我把一个区间更新拆成几个子区间的区间更新，因为是sum，所以对应每个区间都要加上增加的值c*(nr-nl+1),对于找到的区间我就标记，下次考虑这个区间的子区间的时候就要把之前的操作给加上去，并且把这个区间的操作给消0
void update(int nl,int nr,int c,int root)
{
    segtree[root].sum+=c*(nr-nl+1);
    if(nl==segtree[root].l && nr==segtree[root].r)
    {
        segtree[root].c+=c;
        return ;
    }
    if(segtree[root].c)
    {
        segtree[root*2].c+=segtree[root].c;
        segtree[root*2].sum+=segtree[root].c*(segtree[root*2].r-segtree[root*2].l+1);
        segtree[root*2+1].c+=segtree[root].c;
        segtree[root*2+1].sum+=segtree[root].c*(segtree[root*2+1].r-segtree[root*2+1].l+1);
        segtree[root].c=0;
    }
    int mid=(segtree[root].l+segtree[root].r)/2;
    if(nr<=mid) update(nl,nr,c,root*2);
    else if(nl>=mid+1) update(nl,nr,c,root*2+1);
    else
    {
        update(nl,mid,c,root*2);
        update(mid+1,nr,c,root*2+1);
    }
}

void build(int l,int r,int root)
{
    segtree[root].l=l,segtree[root].r=r,segtree[root].c=0;
    if(l==r)
    {
        segtree[root].sum=a[l];
        return ;
    }
    int mid=(l+r)/2;
    build(l,mid,root*2);
    build(mid+1,r,root*2+1);
    segtree[root].sum=segtree[root*2].sum+segtree[root*2+1].sum;
}

int main(void)
{
    int n,q;
    while((scanf("%d%d",&n,&q))==2)
    {
    for(int i=1;i<=n;i++)
        scanf("%lld",&a[i]);
    /*for(int i=1;i<=n;i++)
        printf("%lld",a[i]);*/
    build(1,n,1);
    while(q--)
    {
        ans=0;
        char str[2];
        scanf("%s",str);
        if(str[0]=='C')
        {
            int a,b,c;
            scanf("%d%d%d",&a,&b,&c);
            update(a,b,c,1);
        }
        else if(str[0]=='Q')
        {
            int l,r;
            scanf("%d%d",&l,&r);
            query(l,r,1);
            printf("%lld\n",ans);
        }
    }
    }
}