leetcode560 和为k的子数组-前缀和

题目

给你一个整数数组 nums 和一个整数 k ，请你统计并返回 该数组中和为 k 的连续子数组的个数 。

示例

输入：nums = [1,1,1], k = 2
输出：2

解析

首先要是没思路的话，就用暴力法试试，虽然是一个一维数组，但暴力法还是要循环两次来进行处理，找到符合预期的数据

func subarraySum(nums []int, k int) int {
	n := len(nums)
	res := 0
	for i := 0; i < n; i++ {
		sum := 0 // 放在里面就不用每次置为0了
		for j := i; j < n; j++ {
			sum += nums[j]
			if sum == k {
				res++
			}
		}
	}
	return res
}

前缀和+哈希解法
首先讲一下前缀和的概念，是指通过前缀和数组来保存前n位的和，如presum[1]代表的是nums数组中前1位的和，即

• presum[1] = nums[0]
• presum[2] = nums[0] + nums[1] = presum[1] + nums[1]

以此类推，比如要求nums[2]到nums[4]之间的和，就是：

• presum[5] - presum[2] = nums[2] + nums[3] + nums[4]

因此整个前缀和的公式如下：

	for i := 0; i < n; i++ {
		presum[i+1] = nums[i] + presum[i] // 注意要+1
	}

那么本题目只用前缀和的话，就是如下的代码：

func subarraySum(nums []int, k int) int {
    n := len(nums)
    presum := make([]int, n+1)
    for i := 0; i < n; i++ {
        presum[i+1] = nums[i] + presum[i]
    }

    count := 0
    for i := 0; i < n; i++ {
        for j := i; j < n; j++ {
            if presum[j+1] - presum[i] == k { // 注意下这里比较的是谁和谁
                count++
            }
        }
    }
    return count
}

结果集中有时候会超时，有时候不会，思路是没问题的

需要再加上哈希表来存一下重复的数据。map中存的key是之前的前缀和，value存的是出现的次数，在遍历的时候，判断哈希表中是否存在presum - k，就累加到结果中：
两次遍历方法：

func subarraySum(nums []int, k int) (ans int) {
    s := make([]int, len(nums)+1)
    for i, x := range nums {
        s[i+1] = s[i] + x // 前缀和都需要+1，且默认了s[0] = 0
    }

    cnt := map[int]int{}
    for _, sj := range s { // 遍历的前缀和数组
        ans += cnt[sj-k]
        cnt[sj]++
    }
    return
}

func subarraySum(nums []int, k int) (ans int) {
    s := 0 // 相当于s是前缀和
    cnt := map[int]int{0: 1} // s[0]=0 单独统计
    for _, x := range nums {
        s += x
        ans += cnt[s-k] // 这里的含义是s[i] - s[j] = k 代表有一段和为k
        cnt[s]++
    }
    return
}

func subArraySum3(nums []int, k int) int {
	n := len(nums)
	presum := 0
	count := 0
	tmpMap := make(map[int]int)
	tmpMap[0] = 1
	for i := 0; i < n; i++ {
		presum += nums[i]
		if val, ok := tmpMap[presum-k]; ok {
			count += val
		}
		tmpMap[presum]++
	}
	return count
}