树状数组是利用二进制的一些特点来实现的，主要用来查询动态的连续和的问题，修改和查询都是O(log(n))的复杂度，需要O(n)的空间复杂度
1.单点修改+区间查询
void add(int p, int x){ //给位置p增加x
while(p <= n) sum[p] += x, p += p & -p;
}
int ask(int p){ //求位置p的前缀和
int res = 0;
while(p) res += sum[p], p -= p & -p;
return res;
}
int range_ask(int l, int r){ //区间求和
return ask(r) - ask(l - 1);
}
2.区间修改+单点查询
通过差分，将该问题转化为问题1
差分就是设原数组为a[i]，令数组d[i] = a[i] - a[i-1](a[0] = 0)， 那么a[i] =
∑
j
=
1
i
d
[
j
]
\sum_{j =1}^id[j]
∑j=1id[j]，通过求d[i]的前缀和查询。
那么区间修改就是，当给区间[l,r]加上x的时候，a[l] 与前一个元素 a[l−1] 的差增加了x，a[r+1] 与 a[r] 的差减少了x。根据d[i]数组的定义，只需给d[l] 加上 x, 给d[r+1] 减去 x 即可。
void add(int p, int x){ //这个函数用来在树状数组中直接修改
while(p <= n) sum[p] += x, p += p & -p;
}
void range_add(int l, int r, int x){ //给区间[l, r]加上x
add(l, x), add(r + 1, -x);
}
int ask(int p){ //单点查询
int res = 0;
while(p) res += sum[p], p -= p & -p;
return res;
}
3.区间修改+区间查询
比较常用的部分，也是线段树最难写的一部分。可以用树状数组。
思想也是利用了差分，但是在差分的基础上还用了其他的小技巧。
对于一个位置p的前缀和 ：
∑
i
=
1
p
a
[
i
]
\sum_{i=1}^pa[i]
∑i=1pa[i] =
∑
i
=
1
p
\sum_{i=1}^p
∑i=1p
∑
j
=
1
i
d
[
j
]
\sum_{j=1}^id[j]
∑j=1id[j]。
在右边的等式中d[1]被用了p次， d[2] p-1次……，那么可以将p位置的前缀和写为：
∑
i
=
1
p
\sum_{i=1}^p
∑i=1p
∑
j
=
1
i
d
[
j
]
\sum_{j=1}^id[j]
∑j=1id[j] =
∑
i
=
1
p
d
[
i
]
∗
(
p
−
i
+
1
)
\sum_{i=1}^pd[i]*(p-i+1)
∑i=1pd[i]∗(p−i+1) = (p+1) *
∑
i
=
1
p
d
[
i
]
\sum_{i=1}^pd[i]
∑i=1pd[i] -
∑
i
=
1
p
d
[
i
]
∗
i
\sum_{i=1}^pd[i]*i
∑i=1pd[i]∗i
那么可以维护俩个数组的前缀和：
一个是sum1[i] = d[i]， 另一个是sum2[i] = d[i] * i。
查询
位置p的前缀和即： (p + 1) * sum1数组中p的前缀和 - sum2数组中p的前缀和。
区间[l, r]的和即：位置r的前缀和 - 位置l的前缀和。
修改
对于sum1数组的修改同问题2中对d数组的修改。
对于sum2数组的修改也类似，我们给 sum2[l] 加上 l * x，给 sum2[r + 1] 减去 (r + 1) * x
void add(ll p, ll x){
for(int i = p; i <= n; i += i & -i)
sum1[i] += x, sum2[i] += x * p;
}
void range_add(ll l, ll r, ll x){
add(l, x), add(r + 1, -x);
}
ll ask(ll p){
ll res = 0;
for(int i = p; i; i -= i & -i)
res += (p + 1) * sum1[i] - sum2[i];
return res;
}
ll range_ask(ll l, ll r){
return ask(r) - ask(l - 1);
}
习题
1.luogu p3374【模板】树状数组 1
题意：单点修改，区间查询
代码：
#include <bits/stdc++.h>
using namespace std;
int n, m;
const int maxn = 1e6+50;
int sum[maxn], a[maxn];
void add(int p, int x){
while(p <= n) sum[p] += x, p += p&-p;
}
int ask(int p){
int res = 0;
while(p) res += sum[p], p -= p&-p;
return res;
}
int range_ask(int l, int r){
return ask(r) - ask(l-1);
}
int main()
{
scanf("%d %d", &n, &m);
for(int i = 1; i <= n; i++){
scanf("%d", &a[i]);
add(i, a[i]);
}
for(int i = 1; i <= m; i++){
int op, x, y;
scanf("%d %d %d", &op, &x, &y);
if(op == 1){
add(x, y);
}
else if(op == 2){
printf("%d\n", range_ask(x, y));
}
}
return 0;
}
2.luogu p3368【模板】树状数组 2
链接：luogu p3368
题意：区间修改+单点查询
代码：
#include <bits/stdc++.h>
using namespace std;
#define ll long long
const int maxn = 1e6+50;
int n, m;
ll a[maxn], sum[maxn], b[maxn];
void add(int p, ll x){
while(p <= n) sum[p] += x, p += p&-p;
}
void range_add(int l, int r, ll x){
add(l, x), add(r+1, -x);
}
ll ask(int p){
ll res = 0;
while(p) res += sum[p], p -= p&-p;
return res;
}
int main()
{
scanf("%d %d", &n, &m);
for(int i = 1; i <= n; i++){
scanf("%lld", &a[i]);
b[i] = a[i] - a[i-1];
add(i, b[i]);
}
for(int i = 1; i <= m; i++){
int op, x, y;
ll tmp;
scanf("%d", &op);
if(op == 1){
scanf("%d %d %lld", &x, &y, &tmp);
range_add(x, y, tmp);
}
else if(op == 2){
scanf("%lld", &tmp);
printf("%lld\n", ask(tmp));
}
}
return 0;
}
3.LibreOJ 132. 树状数组 3 ：区间修改，区间查询
题意：区间修改+区间查询
代码：
#include <bits/stdc++.h>
using namespace std;
#define ll long long
const int maxn = 2e6+50;
int n, m;
ll a[maxn], sum1[maxn], sum2[maxn];
void add(ll p, ll x){
for(int i = p; i <= n; i += i & -i)
sum1[i] += x, sum2[i] += x * p;
}
void range_add(ll l, ll r, ll x){
add(l, x), add(r + 1, -x);
}
ll ask(ll p){
ll res = 0;
for(int i = p; i; i -= i & -i)
res += (p + 1) * sum1[i] - sum2[i];
return res;
}
ll range_ask(ll l, ll r){
return ask(r) - ask(l - 1);
}
int main()
{
scanf("%d %d", &n, &m);
for(int i = 1; i <= n; i++){
scanf("%lld", &a[i]);
add(i, a[i]-a[i-1]);
}
for(int i = 1; i <= m; i++){
ll op, x, y;
ll tmp;
scanf("%d", &op);
if(op == 1){
scanf("%lld %lld %lld", &x, &y, &tmp);
range_add(x, y, tmp);
}
else if(op == 2){
scanf("%lld %lld", &x, &y);
printf("%lld\n", range_ask(x, y));
}
}
return 0;
}
4.LibreOJ 10115 校门外的树
链接：LibreOJ 10115
题意：在一段公路上种植树，现在又俩种操作，操作数为1那么表示在接下来读入的l，r区间种上一种全新的树种，（同一个位置可以种多种不同的树），操作数为2表示查询l，r区间的树的种类，注意是种类。
题解：用俩个单点查询+单点修改的树状数组维护一下左括号和右括号a[i]维护左括号，b[i]维护右括号，ans = ask（a，r）- ask（b，l-1）。r左边的左括号表示r右边树的种类，l左边的右括号表示没有在l，r内的树种树。
代码：
#include <bits/stdc++.h>
using namespace std;
int n, m;
const int maxn = 1e6+50;
int a[maxn], b[maxn];
void add(int c[], int p, int x){
while(p <= n) c[p] += x, p += p&-p;
}
int ask(int c[], int p){
int res = 0;
while(p) res += c[p], p -= p&-p;
return res;
}
int main()
{
scanf("%d %d", &n, &m);
for(int i = 1; i <= m; i++){
int op, x, y;
scanf("%d %d %d", &op, &x, &y);
if(op == 1){
add(a, x, 1), add(b, y, 1);
}
else if(op == 2){
printf("%d\n", ask(a, y) - ask(b, x-1));
}
}
return 0;
}
5.POJ 2352 Stars
链接：POj 2352
题意：给你一些星星坐标（x，y），一个星星的等级为左下角（包括相平）即（x0<=x y0 <= y）的星星的个数不包括自己为该星星的等级，现在要你输出0到n-1各个等级星星的个数。
题解：先排个序，按y为第一优先级，x为第二优先级，从小到大排序，然后上树状数组就可以。
代码：
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <cmath>
using namespace std;
int n, m;
const int maxn = 1e5+50;
int ans[maxn], sum[maxn];
struct node{
int x, y;
}a[maxn];
bool cmp(node a, node b){
if(a.y == b.y)
return a.x < b.x;
return a.y < b.y;
}
void add(int p, int x){
while(p <= 40000) sum[p] += x, p += p&-p;
}
int ask(int p){
int res = 0;
while(p) res += sum[p], p -= p&-p;
return res;
}
int main()
{
while(scanf("%d", &n) != EOF){
memset(sum, 0, sizeof(sum));
memset(ans, 0, sizeof(ans));
for(int i = 1; i <= n; i++)
scanf("%d %d", &a[i].x, &a[i].y);
sort(a+1, a+1+n, cmp);
for(int i = 1; i <= n; i++){
ans[ask(a[i].x+1)]++;
add(a[i].x+1, 1);
}
for(int i = 1; i <= n; i++){
printf("%d\n", ans[i-1]);
}
}
return 0;
}
578
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
