golang实现差分

差分可以快速实现数组更新操作,它的主要思想是将对整个数组中一段元素的操作变成差分操作,从而降低计算复杂度。

其实可以将差分看作是前缀和的逆过程,因为差分的核心的思想就是另一个辅助数组计算当前数组相邻两个元素的差值,例如,对于原始数组 [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
	}
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值