树状数组专题

最新推荐文章于 2024-09-18 20:54:10 发布

置顶 DS-K

最新推荐文章于 2024-09-18 20:54:10 发布

阅读量842

点赞数 4

分类专栏：数据结构文章标签：数据结构

本文链接：https://blog.csdn.net/chy20142109/article/details/50673749

版权

数据结构专栏收录该内容

2 篇文章 0 订阅

订阅专栏

树状数组专题

基础和原理就不多说了，做了一些树状数组的题，现在根据题目来大概分一下类。

1、一维单点修改区间查询

这是树状数组最基础的用法，query函数的判断条件如果习惯写成while(x)的请注意一下，如果x为负数的话，会造成死循环。要将数据偏移到1下标开始。

void add(int x,int k)
{
    while(x<=n)
    {
        sum[x]+=k;
        x+=lowbit(x);
    }
}
query(int x)
{
    int ans = 0;
    while(x>0)
    {
        ans+=sum[x];
        x-=lowbit(x);
    }
    return ans;
}

2、一维区间修改单点查询

操作是修改一个区间内的数，查询是只查询一个点。

如果区间范围很大，我们每次模拟修改一遍是非常耗时的。所以我们要优化每次对区间的操作，如果对区间[3,5]操作一次，x[3]++,x[6]--。

那么我们观察前六个数的前缀和为：0 0 1 1 1 0 ，前缀和的状态正好可以表示我们刚才的操作，而且去更新前缀和可以借助树状数组优化至logn。

因此一个位置的前缀和就可以表示这个位置的值。

修改：[x,y] + c

add(x,c) add(y+1,-c)

查询：query(x)

题目： hdu 1556

题解：http://blog.csdn.net/chy20142109/article/details/50674183

3、一维区间修改区间查询

如果对一个区间[x,y]都加上c

那么如果查询前i个数的和：s(i) 表示改变之前的和 s(i)'表示改变之后的和

(1) i>y s(i)’ = s(i) + ( y - x + 1 ) * c

(2) i∈[x,y] s(i)’ = s(i) + ( i - x + 1 ) * c

对于区间修改，我们可以参考第二种，在x位置加上c，y+1位置减去c，这样求得的前缀和的值就只在[x,y]内有效。观察(1)(2)都有一个固定的 (1-x)*c,所以我们可以在x位置先将其减去，再在y+1处加上y*c，用意为，如果范围超过y的话，这个前缀和正好就是( y - x + 1 ) * c。但是如果没超过怎么办？如果在范围内的话c还是有效的，所以加上i*c即可。

这样我们设计两个前缀和来维护。

第一个T1来维护增加量c

第二个T2来维护区间的实际增加量

所以每次操作[x,y]都加上一个值c就可以

T1:add(x,c) add(y+1,-c)

T2:add( x , (1-x)*c ) add( y+1, y*c )

查询[1,x]

T2.query(x) + x * T1.query(x)

如果查询[x,y]，那么就先用查询y的结果减去查询x-1的结果即可。

题目： poj 3468

题解：http://blog.csdn.net/chy20142109/article/details/50674218

4、二维单点修改区间查询

有一个二维矩阵，每次修改一个点，或者查询一个矩形范围内的和。

void add(int x,int y,int k)
{
    while( x <= n )
    {
        int ty = y;
        while( ty <= n )
        {
            sum[x][ty]+=k;
            ty+=lowbit(ty);
        }
        x+=lowbit(x);
    }
}

int query(int x,int y)
{
    int ans = 0;
    while( x>0 )
    {
        int ty = y;
        while(ty>0)
        {
            ans+=sum[x][ty];
            ty-=lowbit(ty);
        }
        x-=lowbit(x);
    }
    return ans;
}

我看到有人用for来写，代码也很简洁。
void add(int x,int y,int k)
{
   for(int i = x; i <= n; i+=lowbit(i))
    for(int j = y; j <= n; j+=lowbit(j))
        sum[i][j]+=k;
}

int query(int x,int y)
{
    int ans = 0;
    for(int i = x; i > 0; i-=lowbit(i))
        for(int j = y; j > 0; j-=lowbit(j))
            ans+=sum[i][j];
    return ans;
}

二维树状数组也是维护一个前缀和，只不过一个增加了一维而已。

回到问题，如果是单点修改的话直接add(x,y,c)即可。

如果是区间查询的话，调用一次query(x,y)显然是求(1,1) 到 (x,y)范围内矩形的和。观察下图，我们没办法一次查询出来给定区间的值，但是我们可以通过计算得出。

现在如果要查询蓝色范围内的和

调用A点是求紫色边框

调用B点是求绿色边框

调用C点是求黄色边框

调用D点是求红色边框

那么A-B-C+D即是答案。