树状数组有两种情况:插点问线和插线问点。这道题是插线问点。
因为树状数组最简单的作用是计算1~x的和,所以给出(a, b, c),表示(a,b)区间增加c, 那我们只需要在a点原来的基础上增加c,然后在b点原来的基础上更新-c,这样我们算最终结果的时候在(a, b)之间的就是增加了c,在区间之外的就是没有增加。
代码:
#include <stdio.h>
#include <string.h>
#define M 1000005
int c[M], m;
int lowbit(int x){
return x&(-x);
}
int getsum(int x){
int sum = 0;
while(x){
sum += c[x];
x -= lowbit(x);
}
return sum;
}
void add(int x, int val){
while(x <= m){
c[x] += val;
x += lowbit(x);
}
}
int main(){
int t;
scanf("%d%d", &t, &m);
memset(c, 0, sizeof(c));
char s[9];
int a, b, c;
while(t --){
scanf("%s", s);
if(s[0] == 'A'){
scanf("%d%d%d", &a, &b, &c);
add(a, c);
add(b+1, -c);
}
else{
scanf("%d", &a);
printf("%d\n", getsum(a));
}
}
}
题目链接: http://acm.nyist.net/JudgeOnline/problem.php?pid=123