题目描述
如题,已知一个数列,你需要进行下面两种操作:
- 将某区间每一个数加上k。
- 求出某区间每一个数的和。
输入格式
第一行包含两个整数 n, m,分别表示该数列数字的个数和操作的总个数。
第二行包含 n 个用空格分隔的整数,其中第 i 个数字表示数列第 i 项的初始值。
接下来 m 行每行包含 3 或 4 个整数,表示一个操作,具体如下:
- 1 x y k:将区间 [x,y] 内每个数加上 k。
- 2 x y:输出区间 [x,y] 内每个数的和。
输出格式
输出包含若干行整数,即为所有操作 2 的结果。
输入输出样例
输入
5 5
1 5 4 2 3
2 2 4
1 2 3 2
2 3 4
1 1 5 1
2 1 4
输出
11
8
20
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 1e5 + 10;
ll f[N];
int n, m;
struct node{
int l, r;
ll sum, lazy;
node(){
l = r = sum = lazy = 0;
}
}a[N*4];
void update(int k) // 更新 k 的sum
{
a[k].sum = a[k*2].sum + a[k*2+1].sum;
}
void build(int k, int l, int r)
{
a[k].l = l, a[k].r = r;
if(l == r){ // 递归到叶节点
a[k].sum = f[l];
return ;
}
int mid = (l + r) / 2;
build(k * 2, l, mid);
build(k * 2 + 1, mid + 1, r);
update(k);
}
void pushdown(int k) // 将点 k 的懒惰标记下传
{
if(a[k].l == a[k].r){
a[k].lazy = 0;
return ;
}
a[k*2].sum += (a[k*2].r - a[k*2].l + 1) * a[k].lazy;
a[k*2+1].sum += (a[k*2+1].r - a[k*2+1].l + 1) * a[k].lazy;
a[k*2].lazy += a[k].lazy;
a[k*2+1].lazy += a[k].lazy;
a[k].lazy = 0;
}
void changeSegment(int k, int l, int r, int x) // 区间修改
{
if(a[k].l == l && a[k].r == r){
a[k].sum += (r - l + 1) * x; // 更新该区间的sum
a[k].lazy += x; // 懒惰标记叠加
return ;
}
if(a[k].lazy) pushdown(k);
int mid = (a[k].l + a[k].r) / 2;
if(r <= mid) changeSegment(k * 2, l, r, x); // 被修改的区间在左区间
else if(l > mid) changeSegment(k * 2 + 1, l, r, x); // 被修改的区间在右区间
else{ // 把修改区间分解成两块,分别往左右区间递归
changeSegment(k * 2, l, mid, x);
changeSegment(k * 2 + 1, mid + 1, r, x);
}
update(k);
}
ll query(int k, int l, int r) // 区间查询
{
if(a[k].lazy) pushdown(k);
if(a[k].l == l && a[k].r == r) return a[k].sum;
int mid = (a[k].l + a[k].r) / 2;
if(r <= mid) return query(k * 2, l, r);
if(l > mid) return query(k * 2 + 1, l, r);
return query(k * 2, l, mid) + query(k * 2 + 1, mid + 1, r);
}
int main()
{
cin >> n >> m;
for(int i = 1; i <= n; i++){
cin >>f[i];
}
build(1, 1, n);
for(int i = 1; i <= m; i++){
int a, b, c, d;
cin >> a;
if(a == 1){
cin >> b >> c >> d;
changeSegment(1, b, c, d);
}
else if(a == 2){
cin >> b >> c;
ll ans = query(1, b, c);
cout << ans << endl;
}
}
return 0;
}