目录
以洛谷P3368为例。首先要明确,这道题采用暴力解法会超时,因此必须用树状数组优化。在笔者的上一篇文章里提到了树状数组解决单点修改和区间求和问题的方法,然而这道题要解决的是区间修改和单点查询,没法使用上一篇文章里的方法,而需要采用差分法。
什么是差分?
令 b[ i ]=c[ i ]-c[ i-1],则c[ i ]=b[ i ]+c[ i-1] =b[ i ]+b[ i-1]+c[ i-2]=……=b[ i ]+b[ i-1]+……+b[1]。
因此,在打代码时我们要构造一个b[ i ]数组,用他来表示题目输入的c[ i ]数组。
这里说的b[ i ]数组实际上就是一个树状数组,我们利用树状数组求前缀和的优越性,以及c[ i ]=前缀和的特性,巧妙的用树状数组某一元素的前缀和来表示题目输入的c[ i ]数组。
差分的作用是什么?
可以发现,运用差分法处理数组之后,成功地构造出了c[ i ]=前缀和的形式,这样就能运用上一篇文章里的方法求解了。
上文提到的b[ i ]其实就是上图的t[ i ],上文提到的c[ i ]就是t[ i ]的前缀和。
代码实现
输入数组数据
输入数据的这段代码要引起重视。由于树状数组的前驱输入的数据会影响到其所有后继存储的值,因此每次输入后面一个节点的值时需要减去上一个节点输入的值,这样才能保证初始的各个树状节点里的数据和题目输入的数据吻合。
操作1
操作1的过程和输入数据的过程很相似,都是需要注意防止节点的后继内的值被改变。
操作2
(sum函数可以改成int型,long long型可能导致莫名错误)
sum()函数就是求某个树状节点的前缀和。
总代码:
#include<iostream>
using namespace std;
int n,c[500010];
int lowbit(int i)
{
return i&(-i);
}
void add(int i,int x)
{
for(;i<=n;i+=lowbit(i))
c[i]+=x;
}
int sum(int j)
{
int sum;
for(;j>0;j-=lowbit(j))
sum+=c[j];
return sum;
}
int main()
{
int m,now,x1,x2,y,k,last=0,a=0;
cin>>n>>m;
for(int i=1;i<=n;i++)
{
cin>>now;
add(i,now-last);
last=now;
}
for(int j=1;j<=m;j++)
{
cin>>a;
if(a==1)
{
cin>>x1>>y>>k;
add(x1,k);
add(y+1,-k);
}
else if(a==2)
{
cin>>x2;
cout<<sum(x2);
if(j!=m)
cout<<endl;
}
}
return 0;
}
/*input
5 5
1 5 4 2 3
1 2 4 2
2 3
1 1 5 -1
1 3 5 7
2 4
*/
/*output
6
10
*/