树状数组总结

树状数组

一、简介

• 解决单点修改，区间查询，单点查询，区间修改等问题的一种数据结构。

• 原理图：
  

二、操作

• lowbit(int x)

  表示最低位1的位置，如$(12{)}_2$=00001100，我们需要取出的是100，此时可以用到前人的智慧，将12对应的二进制取反加一得到11110100，将两者相与即可取出100，因为在计算机中数组的计算使用的是补码，取反加一操作得到的正是原数的相反数，那么我们的lowbit操作即可表示为x&-x.

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

• add(int k,int x)

  修改（加减操作）树状数组中的值，每次修改都要维护树形结构，时间复杂度位$O(logn)$。

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

• getsum(int k)

  获取前k项的和（求前缀和）

  int getsum(int k){
      int ans=0;
      while (k>0)
      {
          ans+=tree[k];
          k-=lowbit(k);
      }
      return ans;
  }
  

• 区间查询

  获取从i到j的总和，也就是前缀和相减。

  int getsum(int i,int j){
  	return getsum(j)-getsum(i-1);
  }
  

• 区间更新、单点查询

  把x~y区间上所有值都加上k（k为实数）

  我们引入差分，可看作前缀和的逆过程，（$D[i]=a[i]-a[i-1]$）,数组a为原始数组，我们利用差分D[i]建树状数组，当我们需要在区间x~y上为每个值加k，那么等价于D[X]+k,D[y+1]-k.

  我们用一个例子说明一下：

  a[]=1 2 3 5 8 9

  那么D[]=1 1 1 2 3 1

  在区间2~4上都加2，a[]=1 4 5 7 10 9 ，那么D[]=1 3 1 2 3 -1，此时我们可以发现D[2]=D[2]’+2,D[6]=D[6]’-2,说明上面总结是正确的。

  单点查询：差分是前缀和的逆过程，那么我们可以得到a是D的前缀和。那么我们想访问a中某个点时就可以用树状数组求其前缀和即可。

  //基本操作均相同
  int main(){
  	for(int i=1;i<=n;i++){
  		cin>>a[i];
  		add(i,a[i]-a[i-1]);//差分
  	}
  	//区间x~y上加k
  	add(x,k);
  	add(y+1,-k);
  	
  	cout<<getsum(i);//单点查询
  	return 0;
  }
  

• 用差分实现区间查询、区间修改
  

  此时我们就需要同时维护两个树状数组。

  #include<iostream>
  using namespace std;
  int n,m,x,y;
  int a[100000] = {0};
  int tree1[100000]={0};    
  int tree2[100000]={0};  
  
  int lowbit(int x){
      return x&(-x);
  }
  
  void add(int i,int k){
      int x = i;    
      while(i <= n){
          tree1[i] += k;
          tree2[i] += k * (x-1);
          i += lowbit(i);
      }
  }
  
  int getsum(int i){       
      int ans = 0, x = i;
      while(i > 0){
          ans += x * tree1[i] - tree2[i];
          i -= lowbit(i);
      }
      return ans;
  }
  
  int main(){
      cin>>n;
      for(int i = 1; i <= n; i++){
          cin>>a[i];
          add(i,a[i] - a[i-1]);   
      }
      int k;
      cin>>x>>y>>k;
      //[x,y]区间内加上k
      add(x,k);   
      add(y+1,-k);      
  
      //求[x,y]区间和
      int sum = getsum(y) - getsum(x-1);
      cout<<sum;
      system("pause");
      return 0;
  }