洛谷-P3372 【模板】线段树 1(区间修改与区间求和)

本文介绍了一种使用线段树进行数组的区间修改和区间求和的基础模板实现。通过详细解释代码,展示了如何构建线段树,进行区间更新以及查询操作。适用于解决涉及大量区间操作的问题。

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

题目链接

题意:

给你一个数组对其进行区间修改和区间求和。

思路:

线段树基础模板。

代码:

#include<bits/stdc++.h>
using namespace std;
#define int long long
#define IOS ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
const int N=1e5+5;
const int inf=0x3f3f3f3f;
int n,m,a[N],ans[N<<2],tag[N<<2];
int ls(int x)
{
    return x*2;
}
int rs(int x)
{
    return x*2+1;
}
void push_up(int p)
{
    ans[p]=ans[ls(p)]+ans[rs(p)];
}
void build(int p,int l,int r)
{
    tag[p]=0;
    if(l==r)
    {
        ans[p]=a[l];
        return ;
    }
    int mid=(l+r)/2;
    build(ls(p),l,mid);
    build(rs(p),mid+1,r);
    push_up(p);
}
void f(int p,int l,int r,int k)
{
    tag[p]=tag[p]+k;
    ans[p]=ans[p]+k*(r-l+1);
}
void push_down(int p,int l,int r)
{
    int mid=(l+r)/2;
    f(ls(p),l,mid,tag[p]);
    f(rs(p),mid+1,r,tag[p]);
    tag[p]=0;
}
void update(int nl,int nr,int l,int r,int p,int k)
{
    if(nl<=l&&r<=nr)
    {
        ans[p]+=k*(r-l+1);
        tag[p]+=k;
        return ;
    }
    push_down(p,l,r);
    int mid=(l+r)/2;
    if(nl<=mid)
    {
        update(nl,nr,l,mid,ls(p),k);
    }
    if(nr>mid)
    {
        update(nl,nr,mid+1,r,rs(p),k);
    }
    push_up(p);
}
int query(int x,int y,int l,int r,int p)
{
    int res=0;
    if(x<=l&&r<=y)
    {
        return ans[p];
    }
    int mid=(l+r)/2;
    push_down(p,l,r);
    if(x<=mid)
    {
        res+=query(x,y,l,mid,ls(p));
    }
    if(y>mid)
    {
        res+=query(x,y,mid+1,r,rs(p));
    }
    return res;
}
signed main()
{
    cin>>n>>m;
    for(int i=1;i<=n;i++)
    {
        cin>>a[i];
    }
    build(1,1,n);
    for(int i=1;i<=m;i++)
    {
        int a,x,y,z;
        cin>>a;
        if(a==1)
        {
            cin>>x>>y>>z;
            update(x,y,1,n,1,z);
        }
        else
        {
            cin>>x>>y;
            cout<<query(x,y,1,n,1)<<endl;
        }
    }
    return 0;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值