【模板】一维前缀和与一维差分

前缀和是一种极其优秀的线性结构,也是一种重要的思想,能极大地降低区间查询的时间复杂度。

为了方便,涉及到前缀和的题目,通常使用全局数组(默认初始化为 0 0 0)且数组下标一般从 1 1 1 开始。

1.一维前缀和

问题描述:假设有一个长度为 n n n 的数组 a [ ] = { a 1 , a 2 , a 3 . . . . . . a n } a[ ] = left{ a_1, a_2, a_3 ...... a_n ight} a[ ]={ a1,a2,a3......an},给出 m m m 次询问,每次询问给出 L , R L, R L,R 两个数,求数组 a [ ] a[ ] a[ ] 在区间 [ L , R ] [L,R] [L,R] 上的和。

如果使用暴力解法,每次都遍历一遍给出的区间,计算出答案,这样时间复杂度会达到 O ( n ∗ m ) O(n*m) O(n∗m),极有可能会 TLE。

如果使用前缀和来做的话,能将时间复杂度降到 O ( n + m ) O(n+m) O(n+m),极大地减少了时间。

1.1 求解一维前缀和数组

一维前缀和数组 s u m [ i ] sum[i] sum[i] 就是原数组中 a [ 1 ] ∼ a [ i ] a[1] sim a[i] a[1]∼a[i] 的总和,代码如下:

void init()
{
          
   
    for (int i = 1; i <= n; i++)
    {
          
   
        sum[i] = sum[i - 1] + a[i];
    }
}

1.2 区间查询

原数组 a [ ] a[ ] a[ ] 在区间 [ L , R ] [L,R] [L,R] 上的和为 sum[R] - sum[L - 1],代码如下:

int get(int L, int R)
{
          
   
    return sum[R] - sum[L - 1];
}

2.一维差分

2.1 定义

假设有原数组 a [ ] = { a 1 , a 2 , a 3 , . . . , a n } a[ ] = left{a_1, a_2, a_3, ... , a_n ight} a[ ]={ a1,a2,a3,...,an},现构造出一个数组 b [ ] = { b 1 , b 2 , b 3 , . . . , b n } b[ ] = left{b_1, b_2, b_3, ... , b_n ight} b[ ]={ b1,b2,b3,...,bn},使得 a i = b 1 + b 2 + . . . + b i a_i=b_1+b_2+...+b_i ai=b1+b2+...+bi,那么 b [ ] b[ ] b[ ] 就称为 a [ ] a[ ] a[ ] 的差分, a [ ] a[ ] a[ ] 就称为 b [ ] b[ ] b[ ] 的前缀和。

可以发现,差分与前缀和是逆运算。

2.2 区间修改

由上述定义可知,差分数组 b [ i ] b[i] b[i] 的前缀和就是原数组 a [ i ] a[i] a[i] 的值。

利用差分数组 b [ ] b[ ] b[ ] 可以快速地对原数组 a [ ] a[ ] a[ ] 进行区间修改,时间复杂度为 O ( 1 ) O(1) O(1)。

假如现在要将原数组 a [ ] a[ ] a[ ] 在区间 [ L , R ] [L,R] [L,R] 上的每个数都加上 x x x,那么通过上述定义可以知道:

    第一个受影响的差分数组中的元素为 b [ L ] b[L] b[L],令 b[L] += x,那么 L L L 及其后面的原数组元素在计算过程中都会加上 x x x。 最后一个受影响的差分数组中的元素为 b [ R ] b[R] b[R],令 b[R + 1] −= x,那么可以保证不会影响 R R R 后面的原数组元素的计算。

这样一来,就不必对区间内每一个数进行处理,只需处理两个端点即可。

void add(int L, int R, int x)
{
          
   
    b[L] += x;
    b[R + 1] -= x;
}

2.3 初始化

问题:一维差分数组 b [ ] b[ ] b[ ] 是如何构造出来的呢?

差分数组 b [ ] b[ ] b[ ] 可以直接通过原数组 a [ ] a[ ] a[ ] 的相邻元素相减得到。

b [ i ] = a [ i ] − a [ i − 1 ] b[i] = a[i] - a[i-1] b[i]=a[i]−a[i−1]

一开始,可以把原数组 a [ ] a[ ] a[ ] 想象成全是 0 0 0,即 a [ ] = { 0 , 0 , 0 , . . . 0 } a[ ] = left{0, 0, 0, ... 0 ight} a[ ]={ 0,0,0,...0},此时相应的差分数组 b [ ] b[ ] b[ ] 也全是 0 0 0, 即 b [ ] = { 0 , 0 , 0 , . . . 0 } b[ ] = left{0, 0, 0, ... 0 ight} b[ ]={ 0,0,0,...0}。

接下来,对原数组 a [ ] a[ ] a[ ] 的初始值可以做如下考虑:

    a [ 1 ] a[1] a[1] 相当于区间 [ 1 , 1 ] [1,1] [1,1] 的每个数都加上 a [ 1 ] a[1] a[1],即 add(1, 1, a[1]) a [ 2 ] a[2] a[2] 相当于区间 [ 2 , 2 ] [2,2] [2,2] 的每个数都加上 a [ 2 ] a[2] a[2],即 add(2, 2, a[2]) … a [ n ] a[n] a[n] 相当于区间 [ n , n ] [n,n] [n,n] 的每个数都加上 a [ n ] a[n] a[n],即 add(n, n, a[n])

这样,利用区间修改操作 a d d ( ) add() add() 即可完成赋初始值,从而避免了手动构造差分数组 b [ ] b[ ] b[ ]。

经验分享 程序员 微信小程序 职场和发展