P3372 【模板】线段树 1

题目描述

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

1. 将某区间每一个数加上 kk。
2. 求出某区间每一个数的和。

输入格式

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

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

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

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

输出格式

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

输入输出样例

输入 #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% 的数据：n \le 8n≤8，m \le 10m≤10。
对于 70\%70% 的数据：n \le {10}^3n≤103，m \le {10}^4m≤104。
对于 100\%100% 的数据：1 \le n, m \le {10}^51≤n,m≤105。

保证任意时刻数列中任意元素的和在 [-2^{63}, 2^{63})[−263,263) 内。

【样例解释】

---

 

#include<iostream>
#include<vector>
#include<queue>
#include<cstring>
#include<cmath>
#include<cstdio>
#include<algorithm>
using namespace std;
const int maxn=1e5+10;
typedef long long LL;
struct SegmentTree{
	LL l,r;
	LL sum,add;
	#define l(x) tree[x].l
	#define r(x) tree[x].r
	#define sum(x) tree[x].sum
	#define add(x) tree[x].add
}tree[maxn*4];
LL a[maxn],n,m;
void build(LL p,LL l,LL r)
{
	l(p)=l;r(p)=r;
	if(l==r) {sum(p)=a[l];return;}
	LL mid=(l+r)/2;
	build(p*2,l,mid);
	build(p*2+1,mid+1,r);
	sum(p)=sum(p*2)+sum(p*2+1);
}
void spread(LL p)
{
	if(add(p))
	{
		sum(p*2)+=add(p)*(r(p*2)-l(p*2)+1);
		sum(p*2+1)+=add(p)*(r(p*2+1)-l(p*2+1)+1);
		add(p*2)+=add(p);add(p*2+1)+=add(p);
		add(p)=0;
	}
}
void change(LL p,LL l,LL r,LL d)
{
	if(l<=l(p)&&r>=r(p)) 
	{
		sum(p)+=(LL)d*(r(p)-l(p)+1);
		add(p)+=d;
		return;
	}
	spread(p);
	LL mid=(l(p)+r(p))/2;
	if(l<=mid) change(p*2,l,r,d);///(l,r)
	if(r>mid) change(p*2+1,l,r,d);
	sum(p)=sum(p*2)+sum(p*2+1);
}
LL ask(LL p,LL l,LL r)
{
   if(l<=l(p)&&r>=r(p)) return sum(p);
   spread(p);
   LL mid=(l(p)+r(p))/2;
   LL val=0;
   if(l<=mid) val+=ask(p*2,l,r);//(l,r)
   if(r>mid) val+=ask(p*2+1,l,r);//(l,r) 
   return val;
}
int main(void)
{
	cin.tie(0);std::ios::sync_with_stdio(false);
	cin>>n>>m;
	for(int i=1;i<=n;i++) cin>>a[i];
	build(1,1,n);
	while(m--)
	{
		int op;LL l,r,d;
		cin>>op;
		if(op==1)
		{
			cin>>l>>r>>d;
			change(1,l,r,d);
		}
		else if(op==2){cin>>l>>r;cout<<ask(1,l,r)<<endl;}
	}
return 0;
}