差分可以快速实现数组更新操作,它的主要思想是将对整个数组中一段元素的操作变成差分操作,从而降低计算复杂度。
其实可以将差分看作是前缀和的逆过程,因为差分的核心的思想就是另一个辅助数组计算当前数组相邻两个元素的差值,例如,对于原始数组 [1, 2, 3, 4, 5]
,其差分数组可以是 [1, 1, 1, 1, 1]。
由此可以看出,差分数组的前缀和数组其实就是原始数组,所以说可以将差分看作是前缀和的逆过程。
而对于差分比较擅长做的事情就是可以对原始数组某个区间快速的进行增减操作。例如如果要对原始数组每个加上1,那时间复杂度就是o(n),而用差分可以做到o(1),那是怎么做到的呢,这和之前的前缀和密不可分,因为原始数组的某个位置的元素等于差分数组的前缀和,假设原始数组中2到4都要加上1,那么只需要差分数组第二个元素加上1就可以实现后面所有元素都加上1,但是我们不需要对5进行操作,故在区间的结束4加上一的后面元素全部再减去加的数1,就可以实现整个区间值的更新。抽象出来既是对于【m,n】m到n这个区间加减c时,将m加上c,将n+1后面的数减去c。再求差分数组的前缀和实现对原始数组的更新。实现代码如下:
func DifferOne(arr []int) {
fmt.Println(arr)
diffArr := make([]int, len(arr))
for i := 0; i < len(arr); i++ {
Insert(diffArr, i, i, arr[i])
}
fmt.Println(diffArr)
//diffArr[0] = arr[0]
//for i := 1; i < len(arr); i++ {
// diffArr[i] = arr[i] - arr[i-1]
//}
Insert(diffArr, 0, 3, 1)
fmt.Println(diffArr)
//求前缀和
prefixSum := make([]int, len(arr))
prefixSum[0] = diffArr[0]
for i := 1; i < len(diffArr); i++ {
prefixSum[i] = prefixSum[i-1] + diffArr[i]
}
fmt.Println(prefixSum)
}
func Insert(arr []int, i, j, c int) {
arr[i] += c
if j < len(arr)-1 {
arr[j+1] -= c
}
}
func DifferTwo(arr [][]int) {
fmt.Println(arr)
diffArr := make([][]int, len(arr))
for i := 0; i < len(arr); i++ {
//这里differarr已经被判别类型了,所以不用:=
diffArr[i] = make([]int, len(arr[i]))
}
for i := 0; i < len(diffArr); i++ {
for j := 0; j < len(diffArr[0]); j++ {
InsertTwo(diffArr, i, j, i, j, arr[i][j])
}
}
fmt.Println(diffArr)
}
func InsertTwo(arr [][]int, j, k, l, m, c int) {
arr[j][k] += c
if k < len(arr)-1 {
arr[k+1][l] -= c
}
if m < len(arr[0])-1 {
arr[j][m+1] -= c
}
if k < len(arr)-1 && m < len(arr[0])-1 {
arr[k+1][m+1] += c
}
}