HDU 1166 敌兵布阵-线段树-（单点更新，区间查询）

标签：hdu   线段树   

题意：n个敌兵的阵营，多个操作：add a ,x：第a个阵营增加x个人；sub a,x：第a个阵营减少x；query a,b：查询区间[a,b]的总人数。

分析：

单点更新，区间查询。直接写。

代码：

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
typedef long long ll;
const int maxn=50000;
int t,n;
ll sum[maxn<<2+10];
void pushup(int rt)
{
	sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}
void creat(int l,int r,int rt)
{
	if(l==r){
		scanf("%d",&sum[rt]);
		return;
	}
	int mid=(l+r)>>1;
	creat(l,mid,rt<<1);
	creat(mid+1,r,rt<<1|1);
	pushup(rt);
}
void upd(int a,int x,int l,int r,int rt)
{
	if(l==r){
		sum[rt]+=x;
		return;
	}
	int mid=(l+r)>>1;
	if(a<=mid) upd(a,x,l,mid,rt<<1);
	else upd(a,x,mid+1,r,rt<<1|1);
	pushup(rt);
}
ll query(int a,int b,int l,int r,int rt)
{
	if(a<=l&&b>=r)  return sum[rt];
	int mid=(l+r)>>1;
	ll ret;
	if(a<=mid) ret=query(a,b,l,mid,rt<<1);
	if(b>mid) ret+=query(a,b,mid+1,r,rt<<1|1);
	return ret;
}
int main()
{
	scanf("%d",&t);
	int cas=1;
	while(t--){
		scanf("%d",&n);
		creat(1,n,1);
		printf("Case %d:\n",cas++);
		char q[20];
		while(scanf("%s",q)!=EOF){
			if(q[0]=='E') break;
			ll a,b;
			scanf("%d%d",&a,&b);
			if(q[0]=='A'||q[0]=='S'){
				if(q[0]=='S') b*=-1;
				upd(a,b,1,n,1);
			}
			else printf("%I64d\n",query(a,b,1,n,1));
		}
	}
}

HDU 1166 敌兵布阵-线段树-（单点更新，区间查询）

标签：hdu   线段树   

原文地址：http://blog.csdn.net/ac_0_summer/article/details/47729521