HDU 1166 敌兵布阵 线段树模板

HDU 1166 敌兵布阵 线段树模板

题目内容是线段树的单点修改和区间查询,可以用作线段树单点更新的模板。

[输入]
第一行一个整数T,表示有T组数据。
每组数据第一行一个正整数N(N<=50000),表示敌人有N个工兵营地,接下来有N个正整数,第i个正整数ai代表第i个工兵营地里开始时有ai个人(1<=ai<=50)。
接下来每行有一条命令,命令有4种形式:
(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 表示结束,这条命令在每组数据最后出现;
每组数据最多有40000条命令

[输出]
对第i组数据,首先输出“Case i:”和回车,
对于每个Query询问,输出一个整数并回车,表示询问的段中的总人数,这个数保持在int以内。

[样例输入]
1
10
1 2 3 4 5 6 7 8 9 10
Query 1 3
Add 3 6
Query 2 7
Sub 10 2
Add 6 3
Query 3 10
End

[样例输出]
Case 1:
6
33
59

AC代码:

#include <iostream>
#include <cstdio> 
#include <string>
#include <cstring>
using namespace std;
int ql,qr,x,d,temp[50005],a[50005*4];
void build(int o,int l,int r)
{
	if(l==r)
		a[o]=temp[l];
	else
	{
		int m=(l+r)/2;
		int ls=2*o,rs=2*o+1;
		build(ls,l,m);
		build(rs,m+1,r);
		a[o]=a[ls]+a[rs];
	}
}//建线段树 
int query(int o,int l,int r)
{
	if(ql<=l&&r<=qr)
		return a[o];
	else
	{
		int m=(l+r)/2;
		int ls=2*o,rs=2*o+1;
		int sum=0;
		if(ql<=m)
			sum+=query(ls,l,m);
		if(qr>=m+1)
			sum+=query(rs,m+1,r);
		return sum;
	}
}//查询 
void update(int o, int l, int r)
{
	if(l==r)
		a[o]+=d;
	else
	{
		int m=(l+r)/2;
		int ls=2*o,rs=2*o+1;
		if(x<=m)
			update(ls,l,m);
		else
			update(rs,m+1,r);
		a[o]=a[ls]+a[rs];
	}
}//修改 
int main()
{
	int t;
	cin>>t;
	for(int T=1;T<=t;T++)
	{
		int n;
		cin>>n;
		for(int i=1;i<=n;i++)
			cin>>temp[i];
		//build(1,1,n);
		memset(a,0,sizeof(a)); 
		for(int i=1;i<=n;i++)
		{
			x=i,d=temp[i];
			update(1,1,n);
		}
		printf("Case %d:\n",T);
		string s;
		int a,b;
		while(cin>>s)
		{
			if(s=="Query")
			{
				cin>>a>>b;
				ql=a,qr=b;
				printf("%d\n",query(1,1,n));
			}
			else if(s=="Add")
			{
				cin>>a>>b;
				x=a,d=b;
				update(1,1,n);
			}
			else if(s=="Sub")
			{
				cin>>a>>b;
				x=a,d=-b;//减去即加上负数 
				update(1,1,n);
			}
			else break;
		}
	}
	return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

球王武磊

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值