树状数组

Posted 2022-06-01 seanocean

 

一、引言

　　1.什么是树状数组？

　　　　顾名思义，就是用数组来模拟树形结构呗。那么衍生出一个问题，为什么不直接建树？答案是没必要，因为树状数组能处理的问题就没必要建树。和Trie树的构造方式有类似之处。

　　2.树状数组可以解决什么问题

　　　　可以解决大部分基于区间上的更新以及求和问题。

　　3.树状数组和线段树的区别在哪里

　　　　树状数组可以解决的问题都可以用线段树解决，这两者的区别在哪里呢？树状数组的系数要少很多，就比如字符串模拟大数可以解决大数问题，也可以解决1+1的问题，但没人会在1+1的问题上用大数模拟。

　　4.树状数组的优点和缺点

　　　　修改和查询的复杂度都是O(logN)，而且相比线段树系数要少很多，比传统数组要快，而且容易写。

　　　　缺点是遇到复杂的区间问题还是不能解决，功能还是有限。

二、基本思想

　　　　算数基本定理：可以将任意正整数关于2的不重复次幂的唯一分解性质，一个数21可以分解为，2^0+2^2+2^4,则一个区间[1,n]可以二进制分解为log（x）个区间

　　　　　　例如：log(21)=3;

　　　　　　　　　　①长度为2^4　　[1,2^4]

　　　　　　　　　　②长度为2^2　　[2^4+1,2^4+2^2]

　　　　　　　　　　③[长度为2^0　　[2^4+2^2+1,2^4+2^2+2^0]

　　　　　树状数组就是依赖算术基本定理的分解思想，把一个区间分解为log（n）个小区间，分而治之

三、基本算法

　　　若区间结尾为R，则区间长度就等于R的二进制分解下最小的二次幂，这时引入lowbit（n）

　　　对于序列A，我们建立一个数组c，数组c保存序列A中[x-lowbit[x]+1,x]中所有数的和

　　　该结构满足一下性质：

• • 　　每个节点c[x]保存以它为根的子树中所有叶子节点的和
  • 　　每个内部节点c[x]的子节点个数等于lowbit（x）的大小
  • 　　除了树根之外，每一个结点c[x]的父亲是c[x+lowbit[x]]
  • 　　树的深度为log(N)

　　1.求lowbit(x)

　　　　原理：将x用二进制表示，将x取反的基础上再加上1，所以原来最后一位到倒着数原来为1的位置（不包含），这些位置原来从0变成1，又因为加上了一，所以就会进位

　　　　　　　原来最后一位到倒着数原来为1的位置（不包含）就会又变成0，直到原来为1的位置与原来是一样的，所以这样就将非负整数x二进制表示下最低位1与后面的0构成的值用lowbit(x)表示

　　　　　　　用因为在补码的表示下~x=-1-n

　　　　　　　所以lowbit(x)=x&(-x);

　　2.对某个元素进行加减法操作

　　　　对a[x]对于修改，c[x]以及c[x]的祖先都需要修改，又因为c[x]的祖先为c[x+lowbit(x)],所以可以在log(n)的时间内执行单点增加以及前缀和维护操作

　　3.查询前缀和

　　　　对于一个前缀和[1,n]，我们把它划分为了log(n)个小区间，想基本思想中举的例子一样，对于区间和，则等于这几个小区间的区间和总值

　　4.查询区间和

　　　　利用前缀和求解，sum[i,j]=sum[j]-sum[i-1]

　　5.扩展(多维树状数组)

　　　　跟一维的差不多，知识多了几个循环，时间复杂度为(logn)^m,在维度不大的情况下，还是可以接受的

　　6.注意事项

　　　　下标不能为0，lowbit(0) ，会陷入死循环

四、典例分析

　　　模板1（单点修改）

#include <bits/stdc++.h>
using namespace std;

int n,m;
int a[50005],c[50005]; //对应原数组和树状数组

int lowbit(int x)
    return x&(-x);


void updata(int i,int k)    //在i位置加上k
    while(i <= n)
        c[i] += k;
        i += lowbit(i);
    


int getsum(int i)        //求A[1 - i]的和
    int res = 0;
    while(i > 0)
        res += c[i];
        i -= lowbit(i);
    
    return res;


int main()
    int t;
    cin>>t;
    for(int tot = 1; tot <= t; tot++)
        cout << "Case " << tot << ":" << endl;
        memset(a, 0, sizeof a);
        memset(c, 0, sizeof c);
        cin>>n;
        for(int i = 1; i <= n; i++)
            cin>>a[i];
            updata(i,a[i]);   //输入初值的时候，也相当于更新了值
        

        string s;
        int x,y;
        while(cin>>s && s[0] != ‘E‘)
            cin>>x>>y;
            if(s[0] == ‘Q‘)    //求和操作
                int sum = getsum(y) - getsum(x-1);    //x-y区间和也就等于1-y区间和减去1-(x-1)区间和
                cout << sum << endl;
            
            else if(s[0] == ‘A‘)
                updata(x,y);
            
            else if(s[0] == ‘S‘)
                updata(x,-y);    //减去操作，即为加上相反数
            
        

    
    return 0;

　　　模板2（区间修改）

int n,m;
int a[50005] = 0;
int sum1[50005];    //(D[1] + D[2] + ... + D[n])
int sum2[50005];    //(1*D[1] + 2*D[2] + ... + n*D[n])

int lowbit(int x)
    return x&(-x);


void updata(int i,int k)
    int x = i;    //因为x不变，所以得先保存i值
    while(i <= n)
        sum1[i] += k;
        sum2[i] += k * (x-1);
        i += lowbit(i);
    


int getsum(int i)        //求前缀和
    int res = 0, x = i;
    while(i > 0)
        res += x * sum1[i] - sum2[i];
        i -= lowbit(i);
    
    return res;


int main()
    cin>>n;
    for(int i = 1; i <= n; i++)
        cin>>a[i];
        updata(i,a[i] - a[i-1]);   //输入初值的时候，也相当于更新了值
    

    //[x,y]区间内加上k
    updata(x,k);    //A[x] - A[x-1]增加k
    updata(y+1,-k);        //A[y+1] - A[y]减少k

    //求[x,y]区间和
    int sum = getsum(y) - getsum(x-1);

    return 0;

五、相关转载与推荐文章(十分感谢这些博主)

　　　　树状数组详解

 

 

 

 

 

l