线段树

最新推荐文章于 2024-04-18 08:24:29 发布

yusiningxin

最新推荐文章于 2024-04-18 08:24:29 发布

阅读量211

点赞数

分类专栏：数据结构

本文链接：https://blog.csdn.net/yusiningxin/article/details/51316430

版权

数据结构专栏收录该内容

1 篇文章 0 订阅

订阅专栏

# include <stdio.h>
# include <string.h>
# include <algorithm>
using namespace std;
const int MAXN=1e5+10;
typedef long long ll;
struct Node
{
    int l,r;
    ll lazy,v;
}t[MAXN<<2];
//区间修改,区间全部加上某个值，区间查询
void push_down (int i) //向下更新
{
    if(t[i].lazy!=0)
    {
        t[i<<1].lazy+=t[i].lazy; //+=
        t[i<<1|1].lazy+=t[i].lazy;
        t[i<<1].v+=(t[i<<1].r-t[i<<1].l+1)*t[i].lazy;
        t[i<<1|1].v+=(t[i<<1|1].r-t[i<<1|1].l+1)*t[i].lazy;
        t[i].lazy=0; //向下更新完之后将自己标记取消
    }
}
void push_up (int i)
{
    t[i].v=t[i<<1].v+t[i<<1|1].v;
}
void build (int l,int r,int i)
{
    t[i].l=l;
    t[i].r=r;
    t[i].lazy=0;
    if(l==r)
    {
        scanf("%lld",&t[i].v);
        return ;
    }
    int m=(l+r)>>1;
    build(l,m,i<<1);
    build(m+1,r,i<<1|1);
    push_up(i);
}
void update (int l,int r,int val,int i) //区间加val
{
    if(t[i].l==l && t[i].r==r)
    {
        t[i].lazy+=val; //lazy标记全为+=，因为有可能上次修改还没有向下更新
        t[i].v+=(r-l+1)*val; //当前节点值一步到位
        return ;
    }
    push_down(i); //不能删除，还不明白为什么
    int m=(t[i].l+t[i].r)>>1;
    if(r<=m) update(l,r,val,i<<1);
    else if(l>=m+1) update(l,r,val,i<<1|1);
    else update(l,m,val,i<<1),update(m+1,r,val,i<<1|1);
    push_up(i); //向上更新值
}
ll query (int l,int r,int i)
{
    if(t[i].l==l&&t[i].r==r) return t[i].v;
    push_down(i); //需要加这个，有的lazy标记可能还没有向下更新操作
    int m=(t[i].l+t[i].r)>>1;
    if(r<=m) return query(l,r,i<<1);
    if(l>=m+1) return query(l,r,i<<1|1);
    return query(l,m,i<<1)+query(m+1,r,i<<1|1);
}
int main ()
{
    //freopen("1.txt","r",stdin);
    int n,q;
    while(scanf("%d%d",&n,&q)!=EOF)
    {
        build(1,n,1);
        while(q--)
        {
            char str[5];
            scanf("%s",str);
            if(str[0]=='Q')
            {
                int a,b;
                scanf("%d%d",&a,&b);
                printf("%lld\n",query(a,b,1));
            }
            else
            {
                int a,b,c;
                scanf("%d%d%d",&a,&b,&c);
                update(a,b,c,1);
            }
        }
    }
    return 0;
}