蓝桥杯学习记录||1264. 动态求连续区间和 || 树状数组_给定一个有n个数的数列,m种操作,规定有两种操作,一是求某个元素的值,二是修改区间-CSDN博客

AcWing||1264. 动态求连续区间和

活动地址：https://www.acwing.com/activity/content/19/

考察要点：数状数组

题目要求

给定 n 个数组成的一个数列，规定有两种操作，一是修改某个元素，二是求子数列 [a,b] 的连续和。

输入格式
第一行包含两个整数 n 和 m，分别表示数的个数和操作次数。

第二行包含 n 个整数，表示完整数列。

接下来 m 行，每行包含三个整数 k,a,b （k=0，表示求子数列[a,b]的和；k=1，表示第 a 个数加 b）。

数列从 1 开始计数。

输出格式
输出若干行数字，表示 k=0 时，对应的子数列 [a,b] 的连续和。

数据范围

1≤n≤100000,
1≤m≤100000，
1≤a≤b≤n

输入样例：

10 5
1 2 3 4 5 6 7 8 9 10
1 1 5
0 1 3
0 4 8
1 7 5
0 4 8

输出样例：

11
30
35

题目地址：https://www.acwing.com/problem/content/1266/

刚开始一来题目都有点难看懂，数状数组对我也是个全新概念。
这道题就是数状数组的经典例题。

解析：

首先是给出一个10个数的数组，然后有两种操作
1.(k = 1) 表示第 a 个数加 b
2. (k = 0)求子数列 [a,b] 的连续和。

1）样例输入 1 1 5 ：k=1 表示操作一让第 1 个数加上 5
那么数组从 1 2 3 4 5 6 7 8 9 10
　　　变成 6 2 3 4 5 6 7 8 9 10

2）样例输入 0 1 3 ：k=0 表示操作二求第 1 个数字到第 3 个数字的和
res = 6 + 2 +3 = 11

快速求某段序列的和，很容易让人想到前缀和，但是前缀和的数组不支持更改，如果有一个数更改了，那么后面的许多数就要重新求了；

对比：

前缀和：区间查询 O(1) 　单点修改：Ｏ(n) 综合:O(n)
数状数组：区间查询O(log n) 单点修改：O(log n) 综合:O(log n)

那么在　修改　和　查询同次数的情况下数状数组的更快一些的

数状数组

在这里插入图片描述
黑笔为原数组，蓝笔为树状数组，连线表示该数为那几个数之和。
由二进制表示的数最低位 1 的值决定层数。
如：5:111 最低位2进制的1为 1
　　 2:10 最低位2进制的10为 2
　　 4:100 最低位的100 为 4
　　 10:1010 最低位的10为 2

lowbit 可以用来计算层数

先讲讲lowbit后面才能继续下去后面还有其他用处

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

其中 -x 其实是x的补码 & 就是一个与操作
lowbit(5) = 101 & 011 = 001 = 1
lowbit(10) = 1010 & 0110 = 0010 = 2

由图可知:

第一层： tr[1] = a[1]; tr[3] = a[3]; …
第二层：tr[2] = a[1] + [2]; tr[6] = a[5] + a[6]; …
第三层：tr[4] = a[1] + a[2] + a[3] + a[4]; …

代码实现是：遍历 a[i] 让对应要加上a[i] 的 tr[]加上
所以操作一：让某个数a[j]加上 x 那么就直接让要加上a[j] 的tr[]加上 x 即可
操作二：求区间和 query(x) 即可求出 a ~ b的区间和

如果求 4~8 的区间和 query(8) - query(4-1)
关键代码 for (int i = x; i; i -= lowbit(i)) res += tr[i];
query(8)

for (int i = 8; i; i -= lowbit(i)) res += tr[i];
res += tr[8]
i-=lowbit(8) => 8 - 8 = 0 —因为 i <=0 结束
query(8) = tr[8]

query(4 - 1)

for (int i = 3; i; i -= lowbit(i)) res += tr[i];
res += tr[3]
i-=lowbit(3) => 3 - 2 = 2 —>res += tr[2]
i-=lowbit(2) => 2 - 2 = 0 —因为 i <=0 结束
query(4 - 1) = tr[3] + tr[2]

query(8) - query(4-1) = tr[8] - tr[3] + tr[2];

就是区间的和，只能说什么，这代码就离谱 Orz

代码思路：输入原数组，求数状数组，根据操作求区间或修改某数

完整代码

#include <iostream>
#include <cstring>
#include <algorithm>
#include <cstdio>
 using namespace std;
const int N = 100010;

int n, m;
int a[N], tr[N];

int lowbit(int x)  // 返回末尾的1 
{
    return x & -x;
}

void add(int x, int v)  
{
    for (int i = x; i <= n; i += lowbit(i)) tr[i] += v;
}

int query(int x)  //求子数列[a,b]的和 //0 1 3 ，求1~3的和 
{
    int res = 0;
    for (int i = x; i; i -= lowbit(i)) res += tr[i];
    return res;
}

int main()
{
    scanf("%d%d", &n, &m);
    for (int i = 1; i <= n; i ++ ) scanf("%d", &a[i]);
    for (int i = 1; i <= n; i ++ ) add(i, a[i]);  //对每个a[i] 让对应的tr[]加上
    
    while (m -- )
    {
        int k, x, y;
        scanf("%d%d%d", &k, &x, &y); //1 1 5
        if(k == 0) printf("%d\n", query(y) - query(x - 1)); //k=0，表示求子数列[a,b]的和
        else add(x, y);    //k=1，表示第 a 个数加 b  ，对应要加上的tr都会加上
    }
    
    return 0;
}