hdu1166敌兵布阵【树形数组的操作:单点更新,区间查询】

题目链接:

http://acm.hdu.edu.cn/showproblem.php?pid=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

思路:

这题正常思路直接去用普通数组去操作的话,必然是超时的下场。
所以,这里就需要用到一个新的知识点。那就是树形数组中的:单点更新,区间查询。
具体这个方法的思路讲解,请参照:https://www.cnblogs.com/xenny/p/9739600.html
这位大佬讲的很仔细了,我在这就不多赘述。

ac代码:

#include <bits/stdc++.h>

using namespace std;

const int maxn = 50005;


int n;

int a[maxn],c[maxn];

//*******树状数据操作**********
int lowbit(int x){
	return x&(-x);
}

void update(int i,int k){
	
	while(i <= n){
		c[i] += k;
		i += lowbit(i);
	}
	
}

int sum(int i){//计算a[1]+...+a[i]的值 
	int ans = 0;
	while(i > 0){
		ans += c[i];
		i -= lowbit(i);
	} 
	return ans;
}
//*********树状数组的操作***********


int main(){
	
	int t;
	int count = 1;//记录到第几个测试用例了 
	cin>>t;
	while(t--){
		cout<<"Case "<<count<<":"<<endl;
		count++;
		cin>>n;
		memset(a,0,sizeof(a));
		memset(c,0,sizeof(c));
		for(int i = 1;i <= n;i++){
			cin>>a[i];
			update(i,a[i]);
		}
		string cmd;
		int a,b;
		int ans;
		while(cin>>cmd){
			if(cmd == "End"){
				break;
			}
			cin>>a>>b;
			if(cmd == "Query"){
				ans = sum(b)-sum(a-1);
				cout<<ans<<endl;
			}else if(cmd == "Add"){
				update(a,b);
			}else if(cmd == "Sub"){
				update(a,-b);
			}
		}
		
		
	}

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值