线段树的模板【区间更新,区间查询】

线段树的模板
线段树讲解

#include <bits/stdc++.h>
#define fill(arr,val) memset(arr,var,sizeof(arr)
using namespace std;
const int maxn=1e5+50;
const int mod=1e9+7;
typedef long long ll;
struct node
{
    int l,r;
    ll sum,lazy;
    void update(ll x)
    {
        sum+=(r-l+1)*x;
        lazy+=x;
    }
}node[4*maxn];
int n,q,a[maxn];
void pushdown(int root)
{
    if(node[root].lazy)
    {
        node[root*2].update(node[root].lazy);
        node[root*2+1].update(node[root].lazy);
        node[root].lazy=0;
    }
}
void pushup(int root)
{
    node[root].sum=node[root*2].sum+node[root*2+1].sum;
}
void build(int root,int l,int r)
{
    node[root].l=l;
    node[root].r=r;
    node[root].lazy=0;
    node[root].sum=0;
    if(l==r)node[root].sum=a[l];
    else
    {
        int mid=l+(r-l)/2;
        build(root*2,l,mid);
        build(root*2+1,mid+1,r);
        pushup(root);
    }
}
void update(int root,int st,int ed,int val)
{
    int l=node[root].l;
    int r=node[root].r;
    if(st>r||ed<l)return;
    if(st<=l&&ed>=r)
    {
        node[root].update(val);
    }
    else
    {
        pushdown(root);
        update(root*2,st,ed,val);
        update(root*2+1,st,ed,val);
        pushup(root);
    }
}
ll query(int root,int st,int ed)
{
    int l=node[root].l;
    int r=node[root].r;
    if(st>r||ed<l)return 0;
    if(st<=l&&ed>=r)return node[root].sum;
    else
    {
        ll ans=0;
        pushdown(root);
        ans+=query(root*2,st,ed);
        ans+=query(root*2+1,st,ed);
        pushup(root);
        return ans;
    }
}
int main()
{
    scanf("%d%d",&n,&q);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&a[i]);
    }
    build(1,1,n);
    for(int i=1;i<=q;i++)
    {
        char ch;
        cin>>ch;
        int x,y,z;
        if(ch=='Q')
        {
            scanf("%d%d",&x,&y);
            printf("%lld\n",query(1,x,y));
        }
        if(ch=='C')
        {
            scanf("%d%d%d",&x,&y,&z);
            update(1,x,y,z);
        }
    }
    return 0;
}
全部评论

相关推荐

评论
点赞
收藏
分享
牛客网
牛客企业服务