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

线段树的模板
线段树讲解

#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;
}
全部评论

相关推荐

但听说转正率很低,我现在有在实习了,好纠结要不要去
熬夜脱发码农:转正率低归低,但是实习的经历你可以拿着,又不是说秋招不准备了
点赞 评论 收藏
分享
不愿透露姓名的神秘牛友
昨天 16:15
你知道对于一个平常不接电话,从来不发语音,只打字交流的人来说电话面有多恐怖吗....刚刚亲眼目睹了舍友电话面...她甚至还在吃饭...就这么水灵灵的打过来开始问了...感觉如果是面对面我真的会紧张到跪下来给面试官磕一个...
一只ikun:额,其实没那么恐怖,最难迈开的是第一步,相信我,你面完第一次后面就不怕了。第一次面试我还想着找个自习室面试,到后面我打着游戏突然来电话我就直接面试了
点赞 评论 收藏
分享
qq乃乃好喝到咩噗茶:院校后面加上211标签,放大加粗,招呼语也写上211
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务