【数据结构】【线段树】单点修改区间查询

#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <algorithm>
using namespace std;

const int MAXSIZE = 50005;

int T,n,x,y,a[MAXSIZE],sum[MAXSIZE*4];
char ask[10];

inline void update(int root)					//上传节点信息 
{
	sum[root]=sum[root<<1]+sum[root<<1|1];
}

void build(int now_root, int l, int r)			//建立线段树 
{
	if (l==r)
	{
		sum[now_root]=a[l];
		return;
	}
	int mid=(l+r)>>1;
	build(now_root<<1,l,mid);
	build(now_root<<1|1,mid+1,r);
	update(now_root);
}

void modify(int i, int l, int r, int x, int d)	//修改:在x处加d 
{
	if (l==r)
	{
		sum[i]+=d;
		return;
	}
	int mid=(l+r)>>1;
	if (x<=mid) modify(i<<1,l,mid,x,d);
	else modify(i<<1|1,mid+1,r,x,d);
	update(i);
}

int query(int i, int l, int r, int L, int R)	//询问L到R区间和 
{
	if (L<=l&&r<=R)
		return sum[i];
	int mid=(l+r)>>1;
	int ans=0;
	if (L<=mid) ans+=query(i<<1,l,mid,L,R);
	if (mid<R) ans+=query(i<<1|1,mid+1,r,L,R);
	return ans;
}

int main()
{
	scanf("%d",&n);
	for(int i=1; i<=n; i++)
		scanf("%d",&a[i]);
	build(1,1,n);		//建立线段树 
	scanf("%s",ask+1);	//读入操作命令:Add,Query,End 
	while(ask[1]!='E')
	{
		scanf("%d%d",&x,&y);
		if (ask[1]=='A')
			modify(1,1,n,x,y);	//在x处加y 
		if (ask[1]=='Q')
			printf("%d\n",query(1,1,n,x,y));	//询问x到y区间和 
		scanf("%s",ask+1);
	}
	return 0;
}


这是HDU1166修改后的模板,用线段树维护区间和,支持单点修改和区间查询操作

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值