线段树模板

最新推荐文章于 2024-06-02 16:48:54 发布

不能AC的小菜鸡

最新推荐文章于 2024-06-02 16:48:54 发布

阅读量62

点赞数

文章标签：算法 c++ 数据结构

本文链接：https://blog.csdn.net/weixin_74139128/article/details/132359665

版权

#include<iostream>
#include<vector>
#include<algorithm>
#include<cstdio>
using namespace std;
const int N=100005;
typedef long long ll;
ll n,m,a[N],tree[N*4],mark[N*4];
inline void push_down(ll p,ll len){
	mark[p*2]+=mark[p];
	mark[p*2+1]+=mark[p];
	tree[p*2]+=mark[p]*(len-len/2);
	tree[p*2+1]+=mark[p]*(len/2);
	mark[p]=0;
}
void build(ll l=1,ll r=n,ll p=1){//建树 
	if(l==r)tree[p]=a[l];
	else {
		ll mid=l+r>>1;
		build(l,mid,p*2);
		build(mid+1,r,p*2+1);
		tree[p]=tree[p*2]+tree[p*2+1];
	}
}
void update(ll l,ll r,ll d,ll p=1,ll cl=1,ll cr=n){
	if(cl>r||cr<l)return;
	else if(cl>=l&&cr<=r){
		tree[p]+=(cr-cl+1)*d;
		if(cr>cl)mark[p]+=d;
	}
	else {
		ll mid=(cl+cr)/2;
		push_down(p,cr-cl+1);
		update(l,r,d,p*2,cl,mid);
		update(l,r,d,p*2+1,mid+1,cr);
		tree[p]=tree[p*2]+tree[p*2+1];
	}
} 
ll query(ll l,ll r,ll p=1,ll cl=1, ll cr=n){
	if(cl>r||cr<l)return 0;
	else if(cl>=l&&cr<=r)return tree[p];
	else {
		ll mid=cl+cr>>1;
		push_down(p,cr-cl+1);
		return query(l,r,p*2,cl,mid)+query(l,r,p*2+1,mid+1,cr);
	}
}
int main(){
	cin>>n>>m;
	for(int i=1;i<=n;i++){
		cin>>a[i];
	}
	build();
	for(int i=0;i<m;i++){
		 ll op,l,r;
		 cin>>op>>l>>r;
		 if(op==1){
		 	ll d;
		 	cin>>d;
		 	update(l,r,d);
		 }
		 else printf("%lld\n",query(l,r));
	}
}