P3374 【模板】树状数组 1

---

原题链接

P3374
题目类型：$普及/提高-$
AC记录：Accepted

题目大意

给你一个数列，你需要下面两种操作：

1.将数列中的某一个数加上$x$
2.求出某区间每一个数的和

输入格式

第一行包含两个正整数$n,m$，分别表示该数列数字的个数和操作的总个数。

第二行包含$n$个用空格分隔的整数，其中第$i$个数字表示数列第$i$项的初始值。

接下来$m$行每行包含$3$个整数，表示一个操作，具体如下：

1 x k含义：将第$x$个数加上$k$
2 x y含义：输出区间$\left[x,y\right]$内每个数的和

输出格式

输出包含若干行整数，即为所有操作$2$的结果。

$\mathbf{S}\mathbf{a}\mathbf{m}\mathbf{p}\mathbf{l}\mathbf{e}$ $\mathbf{I}\mathbf{n}\mathbf{p}\mathbf{u}\mathbf{t}$

5 5
1 5 4 2 3
1 1 3
2 2 5
1 3 -1
1 4 2
2 1 4

$\mathbf{S}\mathbf{a}\mathbf{m}\mathbf{p}\mathbf{l}\mathbf{e}$ $\mathbf{O}\mathbf{u}\mathbf{t}\mathbf{p}\mathbf{u}\mathbf{t}$

14
16

$\mathbf{H}\mathbf{i}\mathbf{n}\mathbf{t}\&\mathbf{E}\mathbf{x}\mathbf{p}\mathbf{l}\mathbf{a}\mathbf{i}\mathbf{n}$

数据范围

对于$30\%$的数据，$1\le n\le 8,1\le m\le 10$。
对于$70\%$的数据，$1\le n,m\le 10^4$。
对于$100\%$的数据，$1\le n,m\le 5\times 10^5$。

解题思路

前置知识

$1.lowbit\text{运算}(不会戳我)$

---

树状数组，是一种不可名状的神奇存储的数的方式。主要用于单点修改和区间查询。

---

存储方式

树状数组只用一个大小为$n$的数组就可以运作，这里设他为$tree\left[1\cdots n\right]$，当然，还有一个存储原来的数的数组$a\left[1\cdots n\right]$。
而$tree\left[x\right]$的含义则是在数组$a$中以第$x$个数为结尾的长度为$lowbit(x)$的区间的和。

---

树状数组的更新

当某一个数被更新后，我们需要更新包含他的所有区间。设被更新的数是$a[x]$，加上了$k$。
我们可以这么想：一个数的$lowbit$运算得到的是这个数最低位的$1$所代表的数，那么加上他就最低位的$1$一定会发生进位。而加上一个$lowbit(x)$在$tree$数组里面只是把区间扩大，所以$tree\left[x+lowbit(x)\right]$这个区间肯定会包含$a[x]$，所以$tree[x+lowbit(x)]$也要更新。
之后只要一直加上去就可以了。
如果对上面的过程不懂，可以参照存储方式里的图来进行理解。
$Examplecode$：

void update(int x,int pos)
{
    while(pos<=n)
    {
        tree[pos]+=x;
        pos+=lowbit(pos);
    }
    return;
}

---

树状数组的区间查询

注：element的翻译为元素。

要解决区间查询，首先要解决一个子问题：查询$a[1\cdots x]$所代表的值的和。
类似上面的图，我们可以把$a[1\cdots x]$分成一个个长度为$2^k$的子区间。
由于树状数组的定义，我们可以直接调用$tree$数组，再相加就可以了。
但问题是：要加哪一个$tree$数组里的元素呢？
由于$lowbit(x)$的值肯定是$2^k$，所以$lowbit(x)$就可以作为子区间的长度。用$x-lowbit(x)$，就可以去掉最低位上的$1$，继续加值。
解决了$a[1\cdots x]$，$a[x\cdots y]$也很简单了，式子为：
$$a[x\cdots y]=a[1\cdots y]-a[1\cdots (x-1)]$$
$Examplecode$：

inline int one_to_n(int pos)
{
    int tar=0;
    while(pos)
    {
        tar+=tree[pos];
        pos-=lowbit(pos);
    }
    return tar;
}

inline int section(int l,int r)
{
    return one_to_n(r)-one_to_n(l-1);
}

---

最后，祝大家早日

上代码

#include<iostream>
using namespace std;

typedef unsigned long long ull;
typedef long long ll;

#define int ll

int tree[500010];
int a[500010];
int n,m;

inline int lowbit(int num) //lowbit运算
{
    return num&(-num);
}

void update(int x,int pos) //更新
{
    while(pos<=n)
    {
        tree[pos]+=x;
        pos+=lowbit(pos);
    }
    return;
}

void init()
{
    for(int i=1; i<=n; i++)
        cin>>a[i],update(a[i],i);
    return;
}

inline int one_to_n(int pos) //a[1...n]
{
    int tar=0;
    while(pos)
    {
        tar+=tree[pos];
        pos-=lowbit(pos);
    }
    return tar;
}

inline int section(int l,int r) //a[l...r]
{
    return one_to_n(r)-one_to_n(l-1);
}

signed main()
{
    cin>>n>>m;
    init();
    for(int i=1; i<=m; i++)
    {
        int x;
        cin>>x;
        if(x==1)
        {
            int y,z;
            cin>>y>>z;
            update(z,y);
        }
        else
        {
            int y,z;
            cin>>y>>z;
            cout<<section(y,z)<<endl;
        }
    }
    return 0;
}

完美切题$\sim$