P3374 【模板】树状数组 1 题解

题目描述

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

• 将某一个数加上 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

说明/提示

【数据范围】

对于 30% 的数据，1≤n≤8，1≤m≤10；
对于 70% 的数据，1≤n,m≤10^4；
对于 100% 的数据，1≤n,m≤5×10^5。

样例说明：

故输出结果14、16

 题解 树状数组板子

#include <bits/stdc++.h>
#ifndef ONLINE_JUDGE
#pragma GCC optimize(2)
//#pragma GCC optimize(3)
#endif
#define INF 0x3f3f3f3f
#define N 500005
#define ll long long
#define ull unsigned long long
#define il inline
#define rg register
using namespace std;
int n, m;
int a[N];
int tr[N];

// lowbit求x的二进制码最低位的1
int lowbit(int x)
{
    //计算机中，正整型的储存码就是其二进制码
    //负数的二进制码为其相反数二进制码的反码+1
    //两个码按位与可得到该数二进制码最低位的1
    return x & -x;
}

//单点加操作
void add(int x, int c)
{
    //将数组中第x位本身及其父节点全部更新，加上c
    for (int i = x; i <= n; i += lowbit(i))
        tr[i] += c;
}

//求a[1...x]区间和操作
int sum(int x)
{
    int res = 0;
    //将属于本区间的数全部相加
    for (int i = x; i; i -= lowbit(i))
        res += tr[i];
    return res;
}
int main()
{
    cin.sync_with_stdio(false);
    cin >> n >> m;
    for (int i = 1; i <= n; i++)
        cin >> a[i];
    //树状数组初始化
    for (int i = 1; i <= n; i++)
    {
        tr[i] += a[i];
        int j = i + lowbit(i);
        if (j <= n)
            tr[j] += tr[i];
    }
    for (int i = 1; i <= m; i++)
    {
        int k, x, y;
        cin >> k >> x >> y;
        if (k == 1)
            add(x, y);
        //求a[x...y]的区间和
        //即为a[1...y]的区间和减去a[1...x-1]的区间和
        if (k == 2)
            cout << sum(y) - sum(x - 1) << endl;
    }
    system("pause");
    return 0;
}