2021-09-26

洛谷P3368 树状数组2

题目描述

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

1. 将某区间每一个数加上 x；
2. 求出某一个数的值。

输入格式

第一行包含两个整数 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;
		}
	}
}