HDU 1166 敌兵布阵 线段树

题意:敌兵阵地上有编号 1 - n 的营地, 每个营地的人员会增加或者减少。题中有四种操作。
(1) Add i j,i和j为正整数,表示第i个营地增加j个人(j不超过30)(2)Sub i j ,i和j为正整数,表示第i个营地减少j个人(j不超过30);(3)Query i j ,i和j为正整数,i<=j,表示询问第i到第j个营地的总人数;(4)End 表示结束,这条命令在每组数据最后出现;
题解:

#include <cstring>
#include <iostream>
using namespace std;

#define L(u) ( u << 1 )
#define R(u) ( u << 1 | 1 )
#define N 50005

int sum[N];
struct item { int l, r, v; } node[N*4];

void build ( int u, int l, int r )
{
	node[u].l = l;
	node[u].r = r;
	node[u].v = sum[r] - sum[l-1];
	if ( l == r ) return;
	int mid = ( l + r ) >> 1;
	build ( L(u), l, mid );
	build ( R(u), mid + 1, r );
}

void update ( int u, int pos, int val )
{
	if ( node[u].l == pos && node[u].r == pos )
	{
		node[u].v += val;
		return;
	}
	int mid = ( node[u].l + node[u].r ) >> 1;
	if ( pos <= mid )
		update ( L(u), pos, val );
	else
		update ( R(u), pos, val );
	node[u].v = node[L(u)].v + node[R(u)].v;
}

int query ( int u, int l, int r )
{
	if ( l <= node[u].l && node[u].r <= r )
		return node[u].v;
	int mid = ( node[u].l + node[u].r ) >> 1;
	if ( r <= mid )
		return query ( L(u), l, r );
	else if ( l > mid )
		return query ( R(u), l, r );
	else
		return query ( L(u), l, mid ) + query ( R(u), mid+1, r );
}

int main()
{
	char oper[10];
	int t, n, i, a, b, num, k = 1;
	scanf("%d",&t);

	while ( t-- )
	{
		sum[0] = 0;
		scanf("%d",&n);
		for ( i = 1; i <= n; i++ )
		{
			scanf("%d",&num);
			sum[i] = sum[i-1] + num;
		}

		build ( 1, 1, n );

		printf("Case %d:\n",k++); 

		while ( scanf("%s",oper) )
		{
			if ( !strcmp(oper,"End") ) break;
			if ( !strcmp(oper,"Query") )
			{
				scanf("%d%d",&a,&b);
				printf("%d\n",query ( 1, a, b ));
			}
			else if ( !strcmp(oper,"Sub") )
			{
				scanf("%d%d",&a,&b);
				update ( 1, a, -b );
			}
			else
			{
				scanf("%d%d",&a,&b);
				update ( 1, a, b );
			}
		}
	}
	return 0;
}


  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值