洛谷P3368 树状数组2
题目描述
如题,已知一个数列,你需要进行下面两种操作:
- 将某区间每一个数加上 x;
- 求出某一个数的值。
输入格式
第一行包含两个整数 N、M,分别表示该数列数字的个数和操作的总个数。
第二行包含 N 个用空格分隔的整数,其中第 i 个数字表示数列第 i 项的初始值。
接下来 M 行每行包含 2 或 4个整数,表示一个操作,具体如下:
操作 1: 格式:1 x y k
含义:将区间 [x,y] 内每个数加上 k;
操作 2: 格式:2 x
含义:输出第 x 个数的值。
输出格式
输出包含若干行整数,即为所有操作 2 的结果。
在树状数组中,c[]是叠加式的,所以在求单点时需要把覆盖该点的区间的值全部加起来
#include <iostream>
using namespace std;
#define lowbit(x) ((x) & (-x))
const int maxn = 500005;
long long c[maxn];
int n, m; //该数列数字的个数,操作的总个数
long long getSum(int x)
{
long long sum = 0;
for (int i = x; i <= n; i += lowbit(i)) //这里的小于n还有疑问
sum += c[i];
return sum;
}
//区间修改
void update(int x, int v)
{
for (int i = x; i > 0; i -= lowbit(i))
c[i] += v;
}
int main(void)
{
cin >> n >> m; //数字的个数,操作的总个数
int op, x, y, k;
for (int i = 1; i <= n; i++)
{
cin >> x;
update(i, x);
update(i - 1, -x);
}
for (int i = 1; i <= m; i++)
{
cin >> op;
if (op == 1)
{
cin >> x >> y >> k;
update(y, k);
update(x - 1, -k);
}
else if (op == 2)
{
cin >> x;
cout << getSum(x) << endl;
}
}
}