/*
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
*/
-
全局变量定义:
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)时标记节点。
-
辅助函数定义:
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 的值。
-
build 函数:
void build(int p, int l, int r)
:建立线段树,当前节点为 p,区间为 [l, r]。- 其中,若 l == r,则将当前节点的值设为 a[l],表示叶节点;否则递归构建左右子树,并通过 push_up 函数更新当前节点值。
-
update 函数:
void update(int ll, int rr, int l, int r, int p, int k)
:更新区间 [ll, rr] 的值为 k,当前操作节点为 p。- 若当前区间包含在更新区间内,则更新当前节点值并标记;否则根据区间位置递归更新左右子树,并通过 push_up 函数更新当前节点值。
-
query 函数:
int query(int x, int y, int l, int r, int p)
:查询区间 [x, y] 的值,当前操作节点为 p。- 若当前区间完全包含在查询区间内,则直接返回节点值;否则根据区间位置递归查询左右子树,并返回结果。
-
主函数 main:
- 读入 n 和 m 以及原始数组 a。
- 调用 build 函数建立线段树。
- 循环 m 次,根据操作类型进行区间更新或查询,更新操作使用 update 函数,查询操作使用 query 函数。