线段树模板(区间修改,区间查询)

题目传送门

给你n个数,m次操作,操作有两种,第一种,对于区间内每个数加上k,第二种输出区间内所有数的和。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=1e6+5;
const int mod=1e9+7;
const int INF=0x7fffffff;
const ll LLINF=0x7fffffffffffffff;
const double EPS=1e-10;
#define IOS ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
#define pb push_back
#define pii pair<int,int>
#define pll pair<ll,ll>
#define debug cout<<"debug"<<endl;
//#define int long long
int a[N];
struct tree
{
    int l,r;
    ll pre,add;
}t[N<<2];
void build(int p,int l,int r)
{
    t[p].l=l;t[p].r=r;
    if(l==r)
    {
        t[p].pre=a[l];
        return ;
    }
    int mid=(l+r)>>1;
    build(p<<1,l,mid);
    build(p<<1|1,mid+1,r);
    t[p].pre=t[p<<1].pre+t[p<<1|1].pre;
}
void spread(int p)
{
    if(t[p].add)
    {
        t[p<<1].pre+=t[p].add*(t[p<<1].r-t[p<<1].l+1);
        t[p<<1|1].pre+=t[p].add*(t[p<<1|1].r-t[p<<1|1].l+1);
        t[p<<1].add+=t[p].add;
        t[p<<1|1].add+=t[p].add;
        t[p].add=0;
    }
}
void change(int p,int x,int y,int z)
{
    if(x<=t[p].l&&t[p].r<=y)
    {
        t[p].pre+=(ll)z*(t[p].r-t[p].l+1);
        t[p].add+=z;
        return ;
    }
    spread(p);
    int mid=(t[p].l+t[p].r)>>1;
    if(x<=mid)
        change(p<<1,x,y,z);
    if(y>mid)
        change(p<<1|1,x,y,z);
    t[p].pre=t[p<<1].pre+t[p<<1|1].pre;
}
ll ask(int p,int x,int y)
{
    if(x<=t[p].l&&y>=t[p].r)
        return t[p].pre;
    spread(p);
    int mid=(t[p].l+t[p].r)>>1;
    ll ans=0;
    if(x<=mid)
        ans+=ask(p<<1,x,y);
    if(y>mid)
        ans+=ask(p<<1|1,x,y);
    return ans;
}
signed main()
{
    IOS;
    //freopen("","r",stdin);
    //freopen("","w",stdout);
    int n,m;
    cin>>n>>m;
    for(int i=1;i<=n;i++)
    {
        cin>>a[i];
    }
    build(1,1,n);
    while(m--)
    {
        int q,x,y,z;
        cin>>q;
        if(q==1)
        {
            cin>>x>>y>>z;
            change(1,x,y,z);
        }
        else
        {
            cin>>x>>y;
            cout<<ask(1,x,y)<<endl;
        }
    }
}

发布了93 篇原创文章 · 获赞 9 · 访问量 4204

猜你喜欢

转载自blog.csdn.net/Joker_He/article/details/104381080