首页 > 其他分享> 文章详细

【模板】线段树1

2021-02-11 20:02:45 阅读：148 来源： 互联网

标签：int 线段 sums mid cin lz ll 模板

17.【模板】线段树1

题目描述

如题，已知一个数列，你需要进行下面两种操作：

将某区间每一个数加上 k。
求出某区间每一个数的和。

输入格式

第一行包含两个整数 n,m，分别表示该数列数字的个数和操作的总个数。

第二行包含 n 个用空格分隔的整数，其中第 ii 个数字表示数列第 ii 项的初始值。

接下来 mm 行每行包含 33 或 44 个整数，表示一个操作，具体如下：

1 x y k：将区间 [x,y] 内每个数加上 k。
2 x y：输出区间 [x,y] 内每个数的和。

输出格式

输出包含若干行整数，即为所有操作 2 的结果。

输入输出样例

输入 #1复制

输出 #1复制

11
8
20

说明/提示

【样例解释】

题解

#include<iostream>
using namespace std;
#define maxn 500005
#define ll long long
typedef struct
{
    int l,r;
    ll sums,lz;
}node;
node a[maxn];
ll c[maxn];
int n,m;
void build(int i,int l,int r)
{
    a[i].l=l,a[i].r=r;
    if(l==r) {
        a[i].sums=c[l];
        return ;}
    int mid=(l+r)>>1;
    build(2*i,l,mid);
    build(2*i+1,mid+1,r);
    a[i].sums=a[i*2].sums+a[i*2+1].sums; 
}

void pushdown(int x)
{
    if(a[x].lz!=0)
    {
        a[2*x].lz+=a[x].lz;
        a[2*x+1].lz+=a[x].lz;
        int mid=(a[x].r+a[x].l)>>1;
        a[2*x].sums+=a[x].lz*(mid-a[2*x].l+1);
        a[2*x+1].sums+=a[x].lz*(a[2*x+1].r-mid);
        a[x].lz=0;
    }
    return ;
}
void update(int i,ll k,int l,int r)
{
    if(l<=a[i].l&&a[i].r<=r)
    {
        a[i].sums+=k*(a[i].r-a[i].l+1);
        a[i].lz+=k;
        return ;
    }
    pushdown(i);
    if(a[2*i].r>=l) update(2*i,k,l,r);
    if(a[2*i+1].l<=r) update(2*i+1,k,l,r);
    a[i].sums=a[2*i].sums+a[2*i+1].sums;
    return ;
}

ll getsum(int i,int l,int r)
{
    if(a[i].l>=l&&a[i].r<=r)
    {
        return a[i].sums;
    }
    ll res=0;
    pushdown(i);
    if(a[2*i].r>=l) res+=getsum(2*i,l,r);
    if(a[2*i+1].l<=r) res+=getsum(2*i+1,l,r);
    return res;
}
int main()
{
    cin>>n>>m;
    for(int i=1;i<=n;i++)
    {
        cin>>c[i];
    }
    build(1,1,n);
    int p1,x,y; ll k;
    for(int i=1;i<=m;i++)
    {
        cin>>p1;
        switch(p1)
        {
            case 1:cin>>x>>y>>k;update(1,k,x,y);break;
            case 2:cin>>x>>y;cout<<getsum(1,x,y)<<endl;break;
            default:break;
        }
    }
    return 0;
}

标签：int,线段,sums,mid,cin,lz,ll,模板
来源： https://blog.csdn.net/lh2018i/article/details/113791619

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

ICode9