[模板]线段树 | 易学教程

题目描述

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

1.将某区间每一个数加上x

2.求出某区间每一个数的和

输入格式

第一行包含两个整数N、M，分别表示该数列数字的个数和操作的总个数。

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

接下来M行每行包含3或4个整数，表示一个操作，具体如下：

操作1：格式：1 x y k 含义：将区间[x,y]内每个数加上k

操作2：格式：2 x y 含义：输出区间[x,y]内每个数的和

输出格式

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

输入输出样例

输入 #1

输出 #1

11
8
20

说明/提示

时空限制：1000ms,128M

数据规模：

对于30%的数据：N<=8，M<=10

对于70%的数据：N<=1000，M<=10000

对于100%的数据：N<=100000，M<=100000

（数据已经过加强^_^，保证在int64/long long数据范围内）

#include<bits/stdc++.h>
using namespace std;
int a[1000],val[1700000],lazy[1700000],m,n,op,v,x,y;
void push_up(int root,int len)
{
    val[root]=val[root<<1]+val[(root<<1)+1]+lazy[root<<1]*(len-len/2)+lazy[root<<1|1]*len/2;
}
void build(int l,int r,int root)
{
    if(l==r)
    {
        scanf("%d",&val[root]);
        return;
    }
    int mid=(l+r)>>1;
    build(l,mid,root<<1);
    build(mid+1,r,(root<<1)+1);
    push_up(root,r-l+1);
}
void update(int l,int r,int root,int pos,int change)
{
    if(l==r)
    {
        val[root]+=change;
        return;
    }
    int mid=(r+l)>>1;
    if(pos<mid) update(l,mid,root<<1,pos,change);
    else update(mid+1,r,(root<<1)+1,pos,change);
    push_up(root,r-l+1);
}
void push_down(int root,int len)
{
    if(!lazy[root]) return;
    val[root]+=len*lazy[root];
    lazy[root<<1]+=lazy[root];
    lazy[(root<<1)+1]+=lazy[root];
    lazy[root]=0;
}
void update_line(int target_l,int target_r,int l,int r,int change,int root)
{
    if(target_l<=l&&r<=target_r)
    {
        lazy[root]+=change;
        push_down(root,r-l+1);
        return;
    }
    push_down(root,r-l+1);
    int mid=(l+r)>>1;
    if(target_l<=mid) update_line(target_l,target_r,l,mid,change,root<<1);
    if(target_r>mid) update_line(target_l,target_r,mid+1,r,change,(root<<1)+1);
    push_up(root,r-l+1);
}
int query(int target_l,int target_r,int l,int r,int root)
{
    if(target_l<=l&&r<=target_r) return val[root]+(r-l+1)*lazy[root];
    push_down(root,r-l+1);
    int mid=(l+r)>>1;
    int ans=0;
    if(target_l<=mid) ans+=query(target_l,target_r,l,mid,root<<1);
    if(target_r>mid) ans+=query(target_l,target_r,mid+1,r,(root<<1)+1);
    return ans;
}
int main()
{
    freopen("testdata.in","r",stdin);
    scanf("%d%d",&n,&m);
    build(1,n,1);
    while(m--)
    {
        scanf("%d",&op);
        if(op==1)
        {
            scanf("%d%d%d",&x,&y,&v);
            update_line(x,y,1,n,v,1);
        }
        else
        {
            scanf("%d%d",&x,&y);
            printf("%d\n",query(x,y,1,n,1));
        }
        
    }
}

来源：https://www.cnblogs.com/satans/p/11947541.html

标签

线段树

root

target