题目链接:
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;
}