求调线段树模板
  • 板块学术版
  • 楼主szlm
  • 当前回复16
  • 已保存回复16
  • 发布时间2024/7/18 09:52
  • 上次更新2024/7/18 10:46:39
查看原帖
求调线段树模板
1159516
szlm楼主2024/7/18 09:52

线段树1【模板】 70tps 后3点RE求调

#include<bits/stdc++.h>
#define int long long  
using namespace std;
const int N=1e5+5;
int n,m,a[N];
struct segtree
{
	int sum;
	int lazyadd;
	int L,R;
}tr[N<<2];
void pushup(int u)
{
	tr[u].sum=tr[u*2].sum+tr[u*2+1].sum;
} 
void pushdown(int u)
{
	if(tr[u].lazyadd == 0) return;
	int mid = (tr[u].L+tr[u].R)/2;
	tr[u<<1].sum += (mid-tr[u].L+1)*tr[u].lazyadd;
	tr[u<<1|1].sum += (tr[u].R-mid)*tr[u].lazyadd; 
	tr[u*2].lazyadd+=tr[u].lazyadd;
	tr[u*2+1].lazyadd+=tr[u].lazyadd;
	tr[u].lazyadd=0;
	return;
}
void build(int l,int r,int u)
{
	tr[u].L=l;
	tr[u].R=r;
	if(l==r) tr[u].sum=a[l];
	else
	{
		int mid=(l+r)>>1;
		build(l,mid,u*2);
		build(mid+1,r,u*2+1);
		tr[u].sum=tr[u*2].sum+tr[u*2+1].sum;
	}
}
void update(int u,int l,int r,int v)
{
	if(l<=tr[u].L&&tr[u].R<=r)
	{
		tr[u].lazyadd+=v;
		tr[u].sum+=(tr[u].R-tr[u].L+1)*v;
	}
	else
	{
		pushdown(u);
		int mid=(tr[u].L+tr[u].R)/2;
		if(l<=mid) update(u*2,l,r,v);
		if(r>mid) update(u*2+1,l,r,v);
		pushup(u);
	}
}
int query(int l,int r,int u)
{
	pushdown(u);
	if(l<=tr[u].L&&tr[u].R<=r) return tr[u].sum;
	int mid=(tr[u].R+tr[u].L)/2;
	int v=0;
	if(l<=mid) v+=query(l,r,u*2);
	if(r>mid) v+=query(l,r,u*2+1);
	return v; 
}
signed main()
{
	cin>>n>>m;
	for(int i=1;i<=n;i++) cin>>a[i];
	build(1,n,1);
	while(m--)
	{
		int t,x,y,k;
		cin>>t>>x>>y;
		if(t==1)
		{
			cin>>k;
			update(1,x,y,k);
		}
		else cout<<query(x,y,1)<<'\n';
	}
}
2024/7/18 09:52
加载中...