洛谷P3372 【模板】线段树 1

P3372 【模板】线段树 1

题目链接-P3372 【模板】线段树 1



解题思路
线段树区间修改+区间查询

• 板子题，没什么说的，具体操作见代码

附上代码

#include<bits/stdc++.h>
#define int long lon
#define lowbit(x) (x &(-x))
#define endl '\n'
using namespace std;
const int INF=0x3f3f3f3f;
const int dir[4][2]={-1,0,1,0,0,-1,0,1};
const double PI=acos(-1.0);
const double e=exp(1.0);
const double eps=1e-10;
const int M=1e9+7;
const int N=1e5+10;
typedef long long ll;
typedef pair<int,int> PII;
typedef unsigned long long ull;
int a[N];
struct node{
	int l,r,sum,lazy;
}t[N<<2];
void push_up(int i){//维护sum值，节点状态更新
	t[i].sum=t[i<<1].sum+t[i<<1|1].sum;
}
void pushdown(int i){//lazy标记下放，维护lazytag
	if(t[i].lazy){
		t[i<<1].sum+=t[i].lazy*(t[i<<1].r-t[i<<1].l+1);
		t[i<<1|1].sum+=t[i].lazy*(t[i<<1|1].r-t[i<<1|1].l+1);
		t[i<<1].lazy+=t[i].lazy;
		t[i<<1|1].lazy+=t[i].lazy;
		t[i].lazy=0;
	}
}
void build(int l,int r,int i){//建树
	//i为当前需要建立的结点，l为当前需要建立区间的左端点，r则为右端点
	t[i].l=l; t[i].r=r;
	if(l==r){//左端点等于右端点，即为叶子节点，直接赋值即可
		t[i].sum=a[l];
		return ;
	}
	int m=l+r>>1;
	build(l,m,i<<1);//构造左儿子节点
	build(m+1,r,i<<1|1);//构造右儿子节点
	push_up(i);//更新父节点
}
void add(int l,int r,int k,int i){//区间修改
	if(l<=t[i].l&&t[i].r<=r){//如果你要访问的区间和你遍历到的这个区间有重合
		t[i].sum+=(t[i].r-t[i].l+1)*k;//那么就直接更新这个区间
		t[i].lazy+=k;//打上lazy标记
		return ;
	}
	pushdown(i);
	int m=t[i].l+t[i].r>>1;
	if(l<=m) add(l,r,k,i<<1);
	if(r>m) add(l,r,k,i<<1|1);
	push_up(i);
}
int query(int l,int r,int i){//区间查询
	if(l<=t[i].l&&t[i].r<=r)//如果当前结点的区间真包含于要查询的区间内，则返回结点信息且不需要往下递归
		return t[i].sum;
	pushdown(i);
	int ans=0;
	int m=t[i].l+t[i].r>>1;
	if(l<=m) ans+=query(l,r,i<<1);//如果左子树和需要查询的区间交集非空
	if(r>m) ans+=query(l,r,i<<1|1);//如果右子树和需要查询的区间交集非空，不能用else,因为查询区间可能同时和左右区间都有交集
	return ans;
}
signed main(){
	ios::sync_with_stdio(false);
	cin.tie(0);cout.tie(0);
	
	int n,m;
	cin>>n>>m;
	for(int i=1;i<=n;i++)
		cin>>a[i];
	build(1,n,1);
	while(m--){
		int op,x,y,k;
		cin>>op>>x>>y;
		if(op==1){
			cin>>k;
			add(x,y,k,1);
		}
		else
			cout<<query(x,y,1)<<endl;
	}
	return 0;
}