题目描述
给定长度为N的数列A,然后输入M行操作指令。
第一类指令形如“C l r d”,表示把数列中第l~r个数都加d。
第二类指令形如“Q X”,表示询问数列中第x个数的值。
对于每个询问,输出一个整数表示答案。
输入格式
第一行包含两个整数N和M。
第二行包含N个整数A[i]。
接下来M行表示M条指令,每条指令的格式如题目描述所示。
输出格式
对于每个询问,输出一个整数表示答案。
每个答案占一行。
思路:
本题说白了就是一个在线单点查询+区间更新,线段树代码太长(懒癌晚期),树状数组可以搞一下,树状数组擅长的是区间查询和单点更新,但是树状数组的本质是前缀和,我们可以让树状数组维护一个差分区间,这样就可以通过单点修改区间的两个端点通过差分区间间接实现区间更新,单点查询只需要对差分区间求前缀和(对应树状数组的区间查询)即可。
code
#include <bits/stdc++.h>
#define mem(a, b) memset(a, b, sizeof a)
#pragma warning (disable:4996)
#pragma warning (disable:6031)
using namespace std;
typedef long long ll;
const int N = 1e5 + 10;
ll n, m;
ll a[N];
int lowbit(int x) {
return x & -x;
}
void add(int x, int d) {
while (x <= n) {
a[x] += d;
x += lowbit(x);
}
}
ll query(int x) {
ll res = 0;
while (x) {
res += a[x];
x -= lowbit(x);
}
return res;
}
int main()
{
mem(a, 0);
scanf("%lld %lld", &n, &m);
ll t;
scanf("%lld", &t);
add(1, t);
for (int i = 2; i <= n; i++) {
ll x;
scanf("%lld", &x);
// 差分数组
add(i, x - t);
t = x;
}
while (m--) {
char ch;
getchar();
ch = getchar();
if (ch == 'Q') {
int x;
scanf("%d", &x);
printf("%lld\n", query(x));
}
else {
int x, y;
ll z;
scanf("%d %d %lld", &x, &y, &z);
add(x, z);
add(y + 1, -z);
}
}
return 0;
}