【模板】线段树 1
题目描述
如题,已知一个数列,你需要进行下面两种操作:
- 将某区间每一个数加上 k k k。
- 求出某区间每一个数的和。
输入格式
第一行包含两个整数 n , m n, m n,m,分别表示该数列数字的个数和操作的总个数。
第二行包含 n n n 个用空格分隔的整数,其中第 i i i 个数字表示数列第 i i i 项的初始值。
接下来 m m m 行每行包含 3 3 3 或 4 4 4 个整数,表示一个操作,具体如下:
1 x y k
:将区间 [ x , y ] [x, y] [x,y] 内每个数加上 k k k。2 x y
:输出区间 [ x , y ] [x, y] [x,y] 内每个数的和。
输出格式
输出包含若干行整数,即为所有操作 2 的结果。
样例 #1
样例输入 #1
5 5
1 5 4 2 3
2 2 4
1 2 3 2
2 3 4
1 1 5 1
2 1 4
样例输出 #1
11
8
20
提示
对于
30
%
30\%
30% 的数据:
n
≤
8
n \le 8
n≤8,
m
≤
10
m \le 10
m≤10。
对于
70
%
70\%
70% 的数据:
n
≤
10
3
n \le {10}^3
n≤103,
m
≤
10
4
m \le {10}^4
m≤104。
对于
100
%
100\%
100% 的数据:
1
≤
n
,
m
≤
10
5
1 \le n, m \le {10}^5
1≤n,m≤105。
保证任意时刻数列中任意元素的和在 [ − 2 63 , 2 63 ) [-2^{63}, 2^{63}) [−263,263) 内。
【样例解释】
思路
线段树模板,每一个根节点表示从l到r的和。
代码
#include<bits/stdc++.h>
using namespace std;
//sum是存线段树的数组,a是输入的数组,add表示l到r区间要增加的树,相当于缓存,如果搜索到他的子节点就会运行q(rt,nl,nr)来更新子节点
long long n,m,sum[4000010],a[4000010],add[4000010];
void p(long long x){
sum[x] = sum[x*2]+sum[x*2+1];
}
void q(long long rt,long long nl,long long nr){
if(add[rt]){
add[rt*2] += add[rt];
add[rt*2+1]+=add[rt];
sum[rt*2]+=add[rt]*nl;
sum[rt*2+1]+=add[rt]*nr;
add[rt] = 0;
}
}
// 初始化函数
void b(long long l,long long r,long long rt){
if(l==r){
sum[rt] = a[l];return ;
}
long long mid = l+r>>1;
b(l,mid,rt*2);
b(mid+1,r,rt*2+1);
p(rt);
}
// 修改区间函数
void xx(long long l,long long r,long long rt,long long x,long long y,long long d){
if(x<=l and y>=r){
sum[rt] +=d*(r-l+1);
add[rt] +=d;
return ;
}
long long mid = l+r>>1;
q(rt,mid-l+1,r-mid);
if(x<=mid) xx(l,mid,rt*2,x,y,d);
if(y>mid) xx(mid+1,r,rt*2+1,x,y,d);
p(rt);
}
// 查询区间函数
long long cc(long long l,long long r,long long rt,long long x,long long y){
if(l>=x and y>=r){
return sum[rt];
}
long long mid = l+r>>1;
long long ans=0;
q(rt,mid-l+1,r-mid);
if(x<=mid){
ans += cc(l,mid,rt*2,x,y);
}
if(y>mid){
ans += cc(mid+1,r,rt*2+1,x,y);
}
return ans;
}
signed main(){
cin >> n >> m;
for(long long i=1;i<=n;i++){
cin >> a[i];
}
b(1,n,1);
for(long long i=1;i<=m;i++){
long long a,b,c,d;
cin >> a;
if(a==1){
cin>>b >> c >> d;
xx(1,n,1,b,c,d);
}else{
cin >> b >> c;
cout << cc(1,n,1,b,c) << endl;
}
}
return 0;
}
感谢收看!!!