目录
题目-一个简单的整数问题2
问题分析
两个操作
- 区间加数值
- 区间查询
首先, 借用一个简单的整数问题思想, 在区间加法 的情况下, 可以将原数组转化为差分数组, 进行区间修改操作
区间查询操作, 可以转化为求前缀和 问题, 因为只要知道了前缀和 , 就知道了当前区间和
假设希望计算 ∑ i = 1 n a i \sum _{i = 1} ^ n a_i ∑i=1nai, 将其用差分序列展开得到
∑ i = 1 n ∑ j = 1 n b j \sum _{i = 1} ^ n \sum _{j = 1} ^ n b_j i=1∑nj=1∑nbj
尝试将上述公式展开为表格, 发现可以将表格补齐, 补齐后的结果
( k + 1 ) ⋅ ( b 1 + b 2 + . . . + b k ) − ( 1 ⋅ b 1 + 2 ⋅ b 2 + . . . + k ⋅ b k ) (k + 1) \cdot (b_1 + b_2 + ... + b_k) - (1 \cdot b_1 + 2 \cdot b_2 + ... + k \cdot b_k) (k+1)⋅(b1+b2+...+bk)−(1⋅b1+2⋅b2+...+k⋅bk)
后面的一段是 i ⋅ b i i \cdot b_i i⋅bi的前缀和
可以使用个树状数组一个维护 b i b_i bi一个维护 i ⋅ b i i \cdot b_i i⋅bi
算法步骤
- 建立两个树状数组
- 初始化树状数组
- 修改操作需要同步修改两个树状数组
- 查询第二个树状数组
因为 A i ≤ 1 0 9 A_i \le 10 ^ 9 Ai≤109, 计算前缀和 需要开 l o n g l o n g long \; long longlong
代码实现
cpp
#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
const int N = 1e5 + 10;
int n, m;
int w[N];
LL tr1[N], tr2[N];
int lowbit(int x) {
return x & -x;
}
void modify(LL tr[], int u, LL val) {
for (int i = u; i <= n; i += lowbit(i)) tr[i] += val;
}
LL get(LL tr[], int u) {
LL ans = 0;
for (int i = u; i; i -= lowbit(i)) ans += tr[i];
return ans;
}
LL calc(int k) {
LL ans = (k + 1) * get(tr1, k) - get(tr2, k);
return ans;
}
int main() {
ios::sync_with_stdio(false);
cin.tie(0);
cin >> n >> m;
for (int i = 1; i <= n; ++i) cin >> w[i];
for (int i = 1; i <= n; ++i) {
modify(tr1, i, (LL) w[i] - w[i - 1]);
modify(tr2, i, (LL) i * (w[i] - w[i - 1]));
}
while (m--) {
char c;
cin >> c;
int l, r, val;
if (c == 'C') {
cin >> l >> r >> val;
modify(tr1, l, val);
modify(tr1, r + 1, -val);
modify(tr2, l, (LL) l * val);
modify(tr2, r + 1, (LL) (r + 1) * -val);
}
else {
cin >> l >> r;
LL ans = calc(r) - calc(l - 1);
cout << ans << '\n';
}
}
return 0;
}