题目大意
给定长度为 \(N\) 的数列 \(A\),然后输入 \(M\) 行操作指令。
第一类指令形如 C l r d
,表示把数列中第 \(l \sim r\) 个数都加 \(d\)。
第二类指令形如 Q x
,表示询问数列中第 \(x\) 个数的值。
对于每个询问,输出一个整数表示答案。
解题思路
\(\qquad\)这题就是建一颗差分树,将区间修改、单点查询转化为单点修改、前缀查询,正好树状数组很适合用来写前缀相关,所以就用树状数组实现。
代码
#include <iostream>
using namespace std;
const int N = 1e5 + 10;
int tr[N], n, Q, a[N];
void add(int x, int v )
{
for (; x <= n; x += x & -x) tr[x] += v;
}
int query(int x)
{
int res = 0;
for (; x; x -= x & -x) res += tr[x];
return res;
}
int main()
{
scanf("%d%d", &n, &Q);
for (int i = 1; i <= n; i ++ )
scanf("%d", &a[i]), add(i, a[i] - a[i - 1]); //构建差分树
while (Q -- )
{
char op[2];
int l, r, d;
scanf("%s%d", op, &l);
if(*op == 'Q') printf("%d\n", query(l)); // 前缀查询
else scanf("%d%d", &r, &d), add(l, d), add(r + 1, -d); // 差分操作
}
return 0;
}