前缀和与差分可以说是特别常用的算法了。今天,就来看看:
- 什么是前缀和与差分
- 前缀和与查分有什么用
正题
定义
我们先来看看前缀和的定义。
对于一个数组 a a a,其前缀数组 s u m sum sum 表示的是 a a a 的一段前缀的数值总和。具体的,有 s u m [ i ] = a [ 1 ] + a [ 2 ] + ⋯ + a [ i ] sum[i]=a[1]+a[2]+\dots+a[i] sum[i]=a[1]+a[2]+⋯+a[i]。
前缀数组最好的求法是递推: s u m [ i ] = s u m [ i − 1 ] + a [ i ] sum[i]=sum[i-1]+a[i] sum[i]=sum[i−1]+a[i]。
那差分是什么呢?
对于 a a a 的差分数组 b b b,有 b [ i ] = a [ i ] − a [ i − 1 ] b[i]=a[i]-a[i-1] b[i]=a[i]−a[i−1]。
并且,前缀数组和差分数组之间有一层惊人的关系:
我们对前缀和递推公式进行移项: s u m [ i ] − s u m [ i − 1 ] = a [ i ] sum[i]-sum[i-1]=a[i] sum[i]−sum[i−1]=a[i]。
所以,如果 s u m sum sum 是 a a a 的前缀数组,那么 a a a 就是 s u m sum sum 的差分数组!!!
同样的,如果 b b b 是 a a a 的差分数组,则 a a a 是 b b b 的前缀数组。
那这有什么用处?
作用
让我们来解决一下这个问题:
给定数组 a a a 和 q q q 次询问,每次询问求区间 [ l , r ] [l,r] [l,r] 中所有数之和。
一样的,暴力肯定不行。但如果我们求出 a a a 的前缀和 s u m sum sum 的话,就简单许多。
我们来看一下这张图:
可以看到,对于 s u m [ r ] sum[r] sum[r],它显然比答案多。但是具体的,它只比答案多出了 s u m [ l − 1 ] sum[l-1] sum[l−1]。所以,此时答案为 s u m [ r ] − s u m [ l − 1 ] sum[r]-sum[l-1] sum[r]−sum[l−1]。这样肯定比暴力优多了。
然而,上述过程中只用到了前缀和。那差分怎么用?
再来看看下面这个题目:
对于一个数组 a a a,有 q q q 次操作,每次操作同时将区间 [ l , r ] [l,r] [l,r] 内的数增加 d d d。 q q q 次操作后,输出 a a a。
这下就可以用到差分数组了。初始时,有 b [ i ] = a [ i ] − a [ i − 1 ] b[i]=a[i]-a[i-1] b[i]=a[i]−a[i−1]。
接下来每次操作,我们考虑不同情况下的 i i i:
- 1 ≤ i < l 1 \leq i<l 1≤i<l 或 r + 1 < i ≤ n r+1<i\leq n r+1<i≤n:此时 b [ i ] = a [ i ] − a [ i − 1 ] → b [ i ] = a [ i ] − a [ i − 1 ] b[i]=a[i]-a[i-1] \to b[i]=a[i]-a[i-1] b[i]=a[i]−a[i−1]→b[i]=a[i]−a[i−1],无变化。
- l < i ≤ r l<i\leq r l<i≤r:此时 b [ i ] = a [ i ] − a [ i − 1 ] → b [ i ] = ( a [ i ] + d ) − ( a [ i − 1 ] + d ) = a [ i ] + d − a [ i − 1 ] − d = a [ i ] − a [ i − 1 ] + ( d − d ) = a [ i ] − a [ i − 1 ] b[i]=a[i]-a[i-1]\to b[i]=(a[i]+d)-(a[i-1]+d)=a[i]+d-a[i-1]-d=a[i]-a[i-1]+(d-d)=a[i]-a[i-1] b[i]=a[i]−a[i−1]→b[i]=(a[i]+d)−(a[i−1]+d)=a[i]+d−a[i−1]−d=a[i]−a[i−1]+(d−d)=a[i]−a[i−1],仍无变化。
- i = l i=l i=l:此时 b [ i ] = a [ i ] − a [ i − 1 ] → b [ i ] = ( a [ i ] + d ) − a [ i − 1 ] = a [ i ] − a [ i − 1 ] + d b[i]=a[i]-a[i-1]\to b[i]=(a[i]+d)-a[i-1]=a[i]-a[i-1]+d b[i]=a[i]−a[i−1]→b[i]=(a[i]+d)−a[i−1]=a[i]−a[i−1]+d, b [ i ] b[i] b[i] 增加 d d d。
- i = r + 1 i=r+1 i=r+1:此时 b [ i ] = a [ i ] − a [ i − 1 ] → b [ i ] = a [ i ] − ( a [ i − 1 ] + d ) = a [ i ] − a [ i − 1 ] − d b[i]=a[i]-a[i-1]\to b[i]=a[i]-(a[i-1]+d)=a[i]-a[i-1]-d b[i]=a[i]−a[i−1]→b[i]=a[i]−(a[i−1]+d)=a[i]−a[i−1]−d, b [ i ] b[i] b[i] 减少 d d d。
因此,当数组 a a a 的区间 [ l , r ] [l,r] [l,r] 所有数都增加 d d d 时,相当于对于其差分数组 b b b, b [ l ] b[l] b[l] 增加 d d d, b [ r + 1 ] b[r+1] b[r+1] 减少 d d d。
例题:
总结:前缀数组和差分数组是可以互相转化的,而我们可以利用这一点巧妙解决各种问题。