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) 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
}