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

线段树的模板
线段树讲解

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

相关推荐

一个菜鸡罢了:哥们,感觉你的简历还是有点问题的,我提几点建议,看看能不能提供一点帮助 1. ”新余学院“别加粗,课程不清楚是否有必要写,感觉版面不如拿来写一下做过的事情,教育经历是你的弱势就尽量少写 2. “干部及社团经历”和“自我评价”删掉 3. 论文后面的“录用”和“小修”啥的都删掉,默认全录用,问了再说,反正小修毕业前肯定能发出来 4. 工作经验和研究成果没有体现你的个人贡献,着重包装一下个人贡献
点赞 评论 收藏
分享
10-11 17:30
湖南大学 C++
我已成为0offer的糕手:羡慕
点赞 评论 收藏
分享
评论
点赞
收藏
分享
牛客网
牛客企业服务