http://acm.nyist.net/JudgeOnline/problem.php?pid=116
树状数组已经看了好几天了,一直都是半懂半不懂,实在是忍无可忍了,今天晚上又看了看刘汝佳的白皮书,似乎明白了树状数组到底是怎么回事,果断写篇博客,明天要把线段树和字典树给搞定,不能再肉了~!
树状数组可以很方便的查询任意区间内所有元素的和,还可以对树进行修改,时间复杂度位log(n)。
有两个很重的数组,A[n] 和 C[n],C 数组保存的是一点连续的和。
对于i 节点,如果他是左孩子,那么父亲 节点的编号为 i + lowbit(i),如果是右孩子,那么父亲节点的编号为i - lowbit(i)。 这里的lowbit(i)指的是i 的最后一个1 对应的值。 如 10 = 1010.则lowbit(10) = 2。
lowbit函数:
int lowbit(int x)
{
return x & (-x);
}
有了C 数组,我们来求前缀Si 的和,妍在节点 i 往左走,边走边向上。把沿途的C[i]都加起来就是前缀和Si了。
sum 函数:
int sum(int x)
{
int num = 0;
while(x > 0)
{
num += c[x];
x -= lowbit(x);
}
return num;
}
同样道理,当修改某个节点的值得时候,是沿着 节点 i 往右走,边走,边向上,把遇到的所有的C[i] 都给修改就行了。
add 函数:
void add(int x, int y)
{
while(x <= N)
{
c[x] += y;
x += lowbit(x);
}
}
附上本题代码:
#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<string.h>
#include<stdlib.h>
#include<algorithm>
using namespace std;
int c[1000100],N;
int lowbit(int x)
{
return x & (-x);
}
int sum(int x)
{
int num = 0;
while(x > 0)
{
num += c[x];
x -= lowbit(x);
}
return num;
}
void add(int x, int y)
{
while(x <= N)
{
c[x] += y;
x += lowbit(x);
}
}
int main()
{
int M,i,j,a,b;
char s[6];
memset(c,0,sizeof(c));
scanf("%d%d",&N,&M);
for(i = 1; i <= N; i++)
{
scanf("%d",&a);
add(i,a);
}
for(i = 0; i < M; i++)
{
scanf("%s%d%d",s,&a,&b);
if(s[0] == 'Q') printf("%d\n",sum(b) - sum(a-1));
else add(a,b);
}
return 0;
}