HDU 1166 敌兵布阵

线段树水题,练习一下线段树的写法

虽然我的写法和网上大多数的写法不同,但感觉我这种写法更好理解,写起来也更简便

单点更新

#include<stdio.h> 
#define N 50005
struct Node{
	int l,r,sum;
}t[N<<2];
int n;
void build(int cur,int l,int r){
	t[cur].l=l;
	t[cur].r=r;
	t[cur].sum=0;
	if(l==r) return;
	int mid=(l+r)>>1;
	build(cur*2,l,mid);
	build(cur*2+1,mid+1,r);
}
void update(int cur,int i,int num) {
	if(t[cur].r<i||t[cur].l>i) return;
	if(t[cur].l<=i&&t[cur].r>=i){
		t[cur].sum+=num;
	}
	if(t[cur].l==t[cur].r) {
		return;
	}
	update(cur*2,i,num);
	update(cur*2+1,i,num);
}
int query(int i,int l,int r){
	if(t[i].r<l||t[i].l>r) return 0;
	if(t[i].l>=l&&t[i].r<=r) {
		return t[i].sum;
	}
	return query(i*2,l,r)+query(i*2+1,l,r);
}
int main(){
	#ifndef ONLINE_JUDGE
	freopen("in.txt","r",stdin);
	#endif
	int t;
	char s[20];
	scanf("%d",&t);
	int cas=1;
	while(t--){
		scanf("%d",&n);
		build(1,1,n);
		int tmp;
		for(int i=1;i<=n;i++) {
			scanf("%d",&tmp);
			update(1,i,tmp);
		}
		printf("Case %d:\n",cas++);
		int tmp1,tmp2;
		while(scanf("%s",s)){
			if(s[0]=='Q'){
				scanf("%d%d",&tmp1,&tmp2);
				printf("%d\n",query(1,tmp1,tmp2));
			}
			else if(s[0]=='A'){
				scanf("%d%d",&tmp1,&tmp2);
				update(1,tmp1,tmp2);
			}
			else if(s[0]=='S'){
				scanf("%d%d",&tmp1,&tmp2);
				update(1,tmp1,-tmp2);
			}
			else break;
		}
	}
	return 0;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值