给定长度为 N 的数列 A,然后输入 M 行操作指令。
第一类指令形如 C l r d
,表示把数列中第 l∼r 个数都加 d。
第二类指令形如 Q x
,表示询问数列中第 x 个数的值。
对于每个询问,输出一个整数表示答案。
输入格式
第一行包含两个整数 N和 M。
第二行包含 N 个整数 A[i]。
接下来 M行表示 M 条指令,每条指令的格式如题目描述所示。
输出格式
对于每个询问,输出一个整数表示答案。
每个答案占一行。
数据范围
1≤N,M≤105,
|d|≤100000,
|A[i]|≤109
输入样例:
10 5
1 2 3 4 5 6 7 8 9 10
Q 4
Q 1
Q 2
C 1 6 3
Q 2
输出样例:
4
1
2
5
思路:
解题思路
这题就是建一颗差分树,将区间修改、单点查询转化为单点修改、区间查询,正好树状数组很适合用来写前缀相关,所以就用树状数组实现。
代码:
#include<bits/stdc++.h>
using namespace std;
#define x first
#define y second
#define endl '\n'
#define IOS ios::sync_with_stdio(false); cin.tie(0);cout.tie(0);
typedef long long LL;
typedef unsigned long long ull;
typedef pair<int, int> PII;
LL lowbit(LL x) { return x & (- x); }
LL gcd(LL a, LL b) { return b ? gcd(b, a % b) : a; }
const int mod = 1e9 + 7;
const int N = 100010;
int n, m;
int a[N];
LL tr[N];
void add(int x, int c)
{
for (int i = x ; i <= n; i += lowbit(i))
tr[i] += c;
}
LL sum(int x)
{
LL res = 0;
for (int i = x; i != 0; i -= lowbit(i))
res += tr[i];
return res;
}
int main()
{
IOS;
cin >> n >> m;
for (int i = 1; i <= n; i++)
cin >> a[i];
for (int i = 1; i <= n; i++)
add(i, a[i] - a[i - 1]);//差分树
while (m--)
{
char op[2];
int l, r, d;
cin >> op >> l;
if (op[0] == 'C')
{
cin >> r >> d;
add(l, d), add(r + 1, -d);
}
else
{
cout << sum(l) << endl;
}
}
return 0;
}