Luogu3372模板线段树

/*
From:Luogu
Problem:【模板】线段树1
Problem uid:3372

URL:https://www.luogu.com.cn/problem/P3372

Time Limit:1000ms
Memory Limit:256Mib

by:CP Editor
*/
#include<bits/stdc++.h>
using namespace std;
const int N=100005;

int n,m;
int a[N];
int tree[N<<2];
int tag[N<<2];

inline int lchild(int p){return p*2;}
inline int rchild(int p){return p*2+1;}

void f(int p,int l,int r,int k){
	tag[p]+=k,tree[p]+=k*(r-l+1);
}

void push_up(int p){tree[p]=tree[lchild(p)]+tree[rchild(p)];}

void push_down(int p,int l,int r){
	int mid=(l+r)>>1;
	f(lchild(p),l,mid,tag[p]),f(rchild(p),mid+1,r,tag[p]);
	tag[p]=0;
}

void build(int p,int l,int r){
	if(l==r)return tree[p]=a[l],void();
	int mid=(l+r)>>1;
	build(lchild(p),l,mid),build(rchild(p),mid+1,r);
	push_up(p);
}

void update(int ll,int rr,int l,int r,int p,int k){
	if(ll<=l&&r<=rr)return tree[p]+=k*(r-l+1),tag[p]+=k,void();
	push_down(p,l,r);
	int mid=(l+r)>>1;
	if(ll<=mid)update(ll,rr,l,mid,lchild(p),k);
	if(rr>mid)update(ll,rr,mid+1,r,rchild(p),k);
	push_up(p);
}

int query(int x,int y,int l,int r,int p){
	int ret=0;
	if(x<=l&&r<=y)return tree[p];
	int mid=(l+r)>>1;
	push_down(p,l,r);
	if(x<=mid)ret+=query(x,y,l,mid,lchild(p));
	if(y>mid)ret+=query(x,y,mid+1,r,rchild(p));
	return ret;	
}

int main(){
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)scanf("%d",&a[i]);
	
	build(1,1,n);
	
	while(m--){
		int op;cin>>op;
		if(op==1){
			int x,y,k;
			scanf("%d%d%d",&x,&y,&k);
			update(x,y,1,n,1,k);
		}else{
			int x,y;
			scanf("%d%d",&x,&y);
			printf("%d\n",query(x,y,1,n,1));
		}
	}
	return 0;
}
/*
记得开longlong，但是我忘开了，也不想改了qwq
*/

1. 全局变量定义：

   • const int N=100005;：定义了一个常量 N，表示数组的最大长度。
   • int n, m;：n 表示数组的元素个数，m 表示操作次数。
   • int a[N];：原始数组。
   • int tree[N<<2];：线段树的存储数组。一般情况下，线段树的空间大小为原数组的 4 倍左右，这里用 N<<2 表示。
   • int tag[N<<2];：标记数组，用于延迟更新（Lazy Propagation）时标记节点。

2. 辅助函数定义：

   • inline int lchild(int p) 和 inline int rchild(int p)：返回左孩子节点和右孩子节点的下标。
   • void f(int p, int l, int r, int k)：更新节点 p 的值和标记。
   • void push_up(int p)：更新节点 p 的值。
   • void push_down(int p, int l, int r)：向下更新节点 p 的值。

3. build 函数：

   • void build(int p, int l, int r)：建立线段树，当前节点为 p，区间为 [l, r]。
   • 其中，若 l == r，则将当前节点的值设为 a[l]，表示叶节点；否则递归构建左右子树，并通过 push_up 函数更新当前节点值。

4. update 函数：

   • void update(int ll, int rr, int l, int r, int p, int k)：更新区间 [ll, rr] 的值为 k，当前操作节点为 p。
   • 若当前区间包含在更新区间内，则更新当前节点值并标记；否则根据区间位置递归更新左右子树，并通过 push_up 函数更新当前节点值。

5. query 函数：

   • int query(int x, int y, int l, int r, int p)：查询区间 [x, y] 的值，当前操作节点为 p。
   • 若当前区间完全包含在查询区间内，则直接返回节点值；否则根据区间位置递归查询左右子树，并返回结果。

6. 主函数 main：

   • 读入 n 和 m 以及原始数组 a。
   • 调用 build 函数建立线段树。
   • 循环 m 次，根据操作类型进行区间更新或查询，更新操作使用 update 函数，查询操作使用 query 函数。