树状数组总结及模板

最新推荐文章于 2023-06-21 22:43:57 发布

角度洗

最新推荐文章于 2023-06-21 22:43:57 发布

阅读量548

点赞数

分类专栏：程序设计思维文章标签： c++ 算法

本文链接：https://blog.csdn.net/ykn37/article/details/125637161

版权

程序设计思维专栏收录该内容

55 篇文章 5 订阅

订阅专栏

树状数组总结及模板

$[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-ij5mYerP-1657080655827)(C:\Users\YYYYYKN\AppData\Roaming\Typora\typora-user-images\image-20220502120625178.png)]$

树状数组的长度就是原数组的长度，树状数组每一个元素的值就是包含该元素的前lowbit()个元素的和。建立的时候可以使用更新代码，边输入数据边更新（反正树状数组最长长度就是输入的所有元素的长度 n，大概率题目是会给的）

lowbit()

#define lb(x) (x & -x)

树状数组查询

$[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-y3p1ELhT-1657080655829)(C:\Users\YYYYYKN\AppData\Roaming\Typora\typora-user-images\image-20220502121302364.png)]$

若是需要查询一段区间 [l, r] 的和，使用 ask® - ask(l - 1) 即可

树状数组更新

只需要修改有包含被修改元素的那几个值就好，也就是把被修改元素下标的二进制的高位一位一位置为1往上走，看图：

$[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-xNg0Giei-1657080655830)(C:\Users\YYYYYKN\AppData\Roaming\Typora\typora-user-images\image-20220502121357352.png)]$

$[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-GtZmtn2Z-1657080655831)(C:\Users\YYYYYKN\AppData\Roaming\Typora\typora-user-images\image-20220503093825888.png)]$

真的注意！！在使用树状数组之前要保证建立好树状数组，而不是直接使用原数组。

例题：

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
#define lb(x) (x & -x)
ll a[10000005] = {0};

void update(ll l, ll v, ll n)//将a[i]加上x，并更新整个树状数组
{
     for (ll i = l; i <= n; i += lb(i))
          a[i] += v;
}

ll qiuhe(ll r)//求下标1-r的和
{
     ll res = 0;
     for (ll i = r; i >= 1; i -= lb(i))
          res += a[i];
     return res;
}
int main()
{
     ll n, q;
     cin >> n >> q;
     for (ll i = 1; i <= n; i++)
     {
          ll t;
          cin >> t;
          update(i, t, n); //初始化树状数组！！！！！！重要步骤，不可缺失！！！！！！！！！
     }
     for (ll i = 0; i < q; i++)
     {
          ll b, c, d;
          cin >> b >> c >> d;
          if (b == 1)
               //更新
               update(c, d, n);
          else
               //求和
               cout << qiuhe(d) - qiuhe(c - 1) << endl;
     }
     return 0;
}