线段树单点修改求助
查看原帖
线段树单点修改求助
162133
_gcl楼主2021/9/11 23:20

30pts

#include<bits/stdc++.h>
using namespace std;
const int N=500000+7; 
inline int read()
{
	int sum=0,f=1;
	char c=getchar();
	while(c<'0'||c>'9')
	{
		if(c=='-')f=-1;
		c=getchar();
	}
	while(c>='0'&&c<='9')
	{
		sum=(sum<<1)+(sum<<3)+(c^48);
		c=getchar();
	}
	return sum*f;
}
struct node
{
	int l,r;
	int v;
}tr[N*4+7];
int n,m,x,j,k;
void build(int u,int l,int r)
{
	tr[u]={l,r};
	if(l==r)return;
	int mid=tr[u].r+tr[u].l>>1;
	build(u<<1,l,mid);
	build(u<<1|1,mid+1,r);
}
void pushup(int u)
{
	tr[u].v=tr[u<<1].v+tr[u<<1|1].v;
}
int query(int u,int l,int r)
{
	if(tr[u].l>=l&&tr[u].r<=r)return tr[u].v;
	int mid=tr[u].l+tr[u].r>>1,v=0;
	if(l<=mid)v=query(u<<1,l,mid);
	if(r>mid)v+=query(u<<1|1,mid+1,r);
	return v;
}
void modify(int u,int x,int v)
{
	if(tr[u].l==x&&tr[u].r==x)tr[u].v+=v;
	else
	{
		int mid=tr[u].r+tr[u].l>>1;
		if(x<=mid)modify(u<<1,x,v);
		else modify(u<<1|1,x,v);
		pushup(u);
	}
}
int main()
{
	n=read(),m=read();
	build(1,1,n);
	for(int i=1;i<=n;i++)
	{
		x=read();
		modify(1,i,x);
	}
	while(m--)
	{
		k=read(),j=read(),x=read();
		if(k==1)
		{
			modify(1,j,x);
		}
		else 
		{
			cout<<query(1,j,x)<<endl;
		}
	}
	return 0;
}
2021/9/11 23:20
加载中...