。。线段树

// Problem: P3372 【模板】线段树 1
// Contest: Luogu
// URL: https://www.luogu.com.cn/problem/P3372
// Memory Limit: 125 MB
// Time Limit: 1000 ms
// 
// Powered by CP Editor (https://cpeditor.org)

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 1e6+ 9;
ll a[N];
//线段树
struct node{
	ll val,tag;//val值,tag懒标记(区间修改)
}seg[N*4];
//区间和合并(汇总)
inline ll tl(ll x){//左叶子
	return x<<1;
}
inline ll tr(ll x){//右叶子
	return x<<1|1;
}
void pushup(const int id){
	seg[id].val=seg[tl(id)].val+seg[tr(id)].val;
}
//建树 O(n)
void build(const int id,int l,int r){
	if(l==r){
		seg[id].val=a[l];
		return;
	}
	int mid=(l+r)>>1;
	build(tl(id),l,mid);
	build(tr(id),mid+1,r);
	pushup(id);
}
//单点查询 id现在在的索引,pos目标索引 O(logn)
ll query1(int id,int l,int r,int pos){
	if(l==r){
		return seg[id].val;
	}else{
		int mid=(l+r)>>1;
		if(mid>=pos){
			return query1(tl(id),l,mid,pos);
		}else{
			return query1(tr(id),mid+1,r,pos);
		}
	}
}
//单点修改 O(logn)
void update1(int id,int l,int r,int pos,ll x){
	if(l==r){
		seg[id].val=x;
	}else{
		int mid=(l+r)>>1;
		if(mid>=pos){
			update1(tl(id),l,mid,pos,x);
		}else{
			update1(tr(id),mid+1,r,pos,x);
		}
		pushup(id);
	}	
}
//O(logn)
//l,r是题目要求区间,L,R是目前区间
//是否包含 T 包含, F 没有包含
bool inrange(int L,int R,int l,int r){
	return ((l<=L) && (R<=r));
}
//有无交集 T 没有交集,F 有交集
bool outofrange(int L,int R,int l,int r){
	return ((L>r)||(R<l));
}
//更新标记
void maketag(int id,int len,ll x){//len,区间修改的长度,x修改值
	seg[id].tag+=x;
	seg[id].val+=(len*x);
}
//汇总标记向下移动
void pushdown(int id,int l,int r){
	int mid=(l+r)>>1;
	maketag(tl(id),mid-l+1,seg[id].tag);
	maketag(tr(id),r-mid,seg[id].tag);
	seg[id].tag=0;//避免重复修改
}
//区间查询
ll query2(int id,int L,int R,int l,int r){
	if(inrange(L,R,l,r)){
		return seg[id].val;
	}else if(!outofrange(L,R,l,r)){
		int mid=(L+R)>>1;//LR不然会越界!
		pushdown(id,L,R);
		return (query2(tl(id),L,mid,l,r)+query2(tr(id),mid+1,R,l,r));
	}else{
		return 0;
	}	
}
//区间修改
void update(int id,int L,int R,int l,int r,ll x){
	if(inrange(L,R,l,r)){
		maketag(id,(R-L+1),x);
	}else if(!outofrange(L,R,l,r)){
		int mid=(L+R)>>1;
		pushdown(id,L,R);
		update(tl(id),L,mid,l,r,x); 
		update(tr(id),mid+1,R,l,r,x);
		pushup(id);	
	}
}
int 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,1,n);
	for(int i=1;i<=m;i++){
		int op;
		cin>>op;
		if(op==1){
			int x,y;
			ll k;
			cin>>x>>y>>k;
			update(1,1,n,x,y,k);
		}else{
			int x,y;
			cin>>x>>y;
			cout<<query2(1,1,n,x,y)<<'\n';
		}
	}
	return 0;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值