前缀和数组

当需要求数组中某连续几项和时可用这个方法,通过先构建前n项和组成新切片,然后通过相减操作即可实现指定连续几项和

package main

import "log"

func main() {
    arr := []int{2, 3, 1, 5, -2}
    log.Println(arr)
    sum := preSumArray(arr)

    l := 1
    r := 3
    log.Println("数组范围[", l, "...", r, "]上累加和:", getSum(sum, l, r))
}

func preSumArray(arr []int) []int {
    n := len(arr)
//构建前n项和新切片
    sum := make([]int, n)
    sum[0] = arr[0]
    for i := 1; i < n; i++ {
        sum[i] = sum[i-1] + arr[i]
    }
    return sum
}

func getSum(sum []int, l, r int) int {
//第一项为0时,可不用相减操作,直接就是前n项和
    if l == 0 {
        return sum[r]
    } else {
        return sum[r] - sum[l-1]
    }
}

时间复杂度为O(1)

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值