normal版本的树状数组模板可完成两种操作(单次都是O(logn)):
①单点修改②区间求和
那么如果题目变成了基础线段树的要求:
①区间修改②单点询问
事实上树状数组也是可以优雅地解决这个问题的,而且代码量的确低得惊人
这需要我们注意到一个无脑而有趣的等式
a
n
=
∑
1
n
(
a
i
−
a
i
−
1
)
a_{n}=\sum_{1}^{n}(a_{i}-a_{i-1})
an=∑1n(ai−ai−1)(不妨让
a
0
=
0
a_{0}=0
a0=0)
这种操作有个看似高贵的名字——差分
那么提示至此,相信读者也能很快地想出该题的做法了
读入时差分存储 b i = a i − a i − 1 b_{i}=a_{i}-a_{i-1} bi=ai−ai−1,询问时用 b n b_{n} bn的前i项和代表 a i a_{i} ai,修改时增加 b l b_{l} bl的值,减少 b r + 1 b_{r+1} br+1的值
附上洛谷原题
题目描述 如题,已知一个数列,你需要进行下面两种操作:
1.将某区间每一个数数加上x
2.求出某一个数的值
输入输出格式 输入格式: 第一行包含两个整数N、M,分别表示该数列数字的个数和操作的总个数。
第二行包含N个用空格分隔的整数,其中第i个数字表示数列第i项的初始值。
接下来M行每行包含2或4个整数,表示一个操作,具体如下:
操作1: 格式:1 x y k 含义:将区间[x,y]内每个数加上k
操作2: 格式:2 x 含义:输出第x个数的值
输出格式: 输出包含若干行整数,即为所有操作2的结果。
https://www.luogu.org/problemnew/show/P3368
AC代码:
#include <bits/stdc++.h>
using namespace std;
const int MAXN=5e5+10;
int n,m,k,x,y,z,a[MAXN],tree[MAXN];
int lowbit(int x) //get the lowest "1" in x(binary form) eg. lowbit(1100(2))=100(2)=4
{
return x & -x;
}
void update(int i,int x)
{
for (;i<=n;tree[i]+=x, i+=lowbit(i));
}
long long sum(int i)
{
long long ret=0;
for (;i;ret+=tree[i], i-=lowbit(i));
return ret;
}
int main()
{
cin>>n>>m;
a[0]=0;
for (int i=1;i<=n;i++)
{
scanf("%d",&a[i]);
update(i,a[i]-a[i-1]);
}
for (int i=1;i<=m;i++)
{
scanf("%d%d",&k,&x);
if (k-1) printf("%lld\n",sum(x));
else
{
scanf("%d%d",&y,&z);
update(x,z);
update(y+1,-z);
}
}
return 0;
}