洛谷 P3374 【模板】树状数组

题目描述

如题，已知一个数列，你需要进行下面两种操作：

• 将某一个数加上 x

• 求出某区间每一个数的和

输入格式

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

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

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

• 1 x k 含义：将第 x 个数加上 k

• 2 x y 含义：输出区间 [x,y] 内每个数的和

输出格式

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

输入输出样例

输入 #1

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

输出 #1

14
16

---

很明显，要做出这道题我们要先学会树状数组，那么什么是树状数组呢？

只要你脑海里面能呈现出树形的结构基本上都是nlogn级别的，这道题用前缀和数组会超时是因为前缀和查询虽然是O(1)但是修改区间是O(n)，这样都超时了那也只能考虑更好的结构了。

  概念  简单理解

二叉树是以一个父亲和一个左儿子和右儿子为规律建立的树

那么简单理解，树状数组是以lowbit()为规律建立的数组

什么是lowbit

lowbit函数里面只有一个int参数，返回的是该int参数二进制下排在最后的1。比如3(011) lowbit(3)=1，4(0100) lowbit(4)=4直接把取出来的1变成十进制就完事了。

//朴素算法
int lowbit(int x){
    int idx=0;
    while(x){
        if(x&1)return pow(2,idx);//如果最后一位是1，返回2^idx
        x>>=1;//整体往右移动1个挤掉最后一位(位运算)
        idx++;//权值计数
    }
}

计算好慢啊，有没有更快的算法

int lowbit(int x){
    return x&-x;
}
假设x为十进制正整数，则-x为十进制负整数
正整数在电脑里面以原码形式保存，负数则以补码(取反加1)的形式保存

假设十进制正整数x的二进制形式为
a1 a2 a3 a4 ..... an ... an+1

易得-x的二进制形式为
~a1 ~a2 ~a3 ~a4 ..... ~an ... ~an+1 +1

任意假设a1~an+1的值
易得永远取到最后一个1

---

知道了什么是lowbit那么我们以lowbit为基础建立一个树状数组

按照lowbit方式建立树状数组

第一层 1 0001 缺3补齐 3 0011   缺5补齐 5 0101

第二层 2 0010         4 0100          6 0110

第三层 4 0100         8 1000          8 1000

第四层 8 1000

按层次一层一层往上画（相同数字取最高层），按照相邻层的数字画线

图片取自 树状数组详解_萧何山的博客-CSDN博客_树状数组

既然是按照lowbit()建立的树，那么只要按照lowbit()的规律就可以遍历整个树

故1+=lowbit(1)等于2 即找到了父节点，2-=lowbit(2)等于1 即找到了儿子节点

那为什么4+=lowbit(4)等于0呢？ 他没有儿子吗？

让我们看看建立树状数组的地方，1，2，3，4的上头都是C[4]，那么对A[1]，A[2]，A[3]，A[4]进行值修改，C[4]都会变化。即C[4]=A[1]+A[2]+A[3]+A[4]。

所以不用去找4的儿子我们就只能能取到A[1]~A[4]的区间和。

---

这只是明确树状数组的用法，深度理解移步 登录 - 洛谷

请注意，树状数组的下标是从1开始的

AC代码（单点修改+区间查询）

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

const int MAXSIZE=5e+5;
int n,m,op,x,y,c[MAXSIZE+1],tmp;

int lowbit(int k){
    return k&-k;
}
int intervalSum(int i){
    int sum=0;
    while(i>=1){
        sum+=c[i];
        i-= lowbit(i);
    }
    return sum;
}
void add(int i,int k){
    while(i<=n){
        c[i]+=k;
        i+= lowbit(i);
    }
}
int main(){
    cin>>n>>m;
    for(int i=1;i<=n;i++){
        scanf("%d",&tmp);
        add(i,tmp);
    }

    for(int i=0;i<m;i++){
        cin>>op>>x>>y;
        if(op==1){
            add(x,y);
        }else{
            cout<<intervalSum(y)- intervalSum(x-1)<<endl;
        }
    }
    return 0;
}

那么有了单点修改和区间查询，有没有区间修改和单点查询呢

树状数组要这么多功能干什么，你以为你是线段树啊

然后我们拿这个模板和单差分去提交第二道题（即用单点修改和区间查询 提交 区间修改和单点查询）

单差分算法 

单模板

模板超时是因为只能单点修改，差分超时是因为单点查询要遍历一次，即O(n)

那糅合一下就能过了。

AC代码（区间修改+单点查询）

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

const int MAXN=5e+5;
int n,m,c[MAXN+1],t;

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

int main(){
    cin>>n>>m;
    for(int i=1;i<=n;i++){
        scanf("%d",&t);
        add(i,t);
        add(i+1,-t);
    }

    int op,x,y,k;
    for(int i=0;i<m;i++){
        scanf("%d",&op);
        if(op==1){
            scanf("%d %d %d",&x,&y,&k);
            add(x,k);
            add(y+1,-k);
        }else{
            scanf("%d",&x);
            cout<<query(x)<<endl;
        }
    }
    return 0;
}

 因为是差分维护的，所以询问就变成了单点查询。