树状数组作为一种数据结构,很广泛的运用到关于统计的问题中,树状数组通过类似数的结构每一位都记录的是从前面到这个点所有的和,要修改起来也很方便。在代码上树状数组也很简单,就是核心的三个函数。
const int maxn = 1002;//一维树状数组
int c[maxn];
int Lowbit(int x)
{
return x&(-x);
}
void update(int x,int detal)
{
while(x <= maxn)
{
c[x] += detal;
x += Lowbit(x);
}
}
int getsum(int x)
{
int sum =0;
while(x > 0)
{
sum += c[x];
x -= Lowbit(x);
}
return sum;
}
同样的,树状数组的应用也可以扩展到二维上,记录从第一个点到目前点的对角线内矩阵的所有点的和。
const int maxn = 1002;//二维树状数组
int c[maxn][maxn];
int Lowbit(int x)
{
return x&(-x);
}
void update(int x,int y,int d)
{
int i,j;
for(i=x; i<maxn; i+=Lowbit(i))
for(j=y; j<maxn; j+=Lowbit(j))
c[i][j] += d;
}
int sum(int x,int y)
{
int sum = 0,i,j;
for(i=x; i>0; i-=Lowbit(i))
for(j=y; j>0; j-=Lowbit(j))
sum += c[i][j];
return sum;
}