Leetcode 560. 和为 K 的子数组前缀和哈希

地址 https://leetcode-cn.com/problems/subarray-sum-equals-k/

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

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

示例 2：
输入：nums = [1,2,3], k = 3
输出：2
 
提示：
1 <= nums.length <= 2 * 10^4
-1000 <= nums[i] <= 1000
-10^7 <= k <= 10^7

解答
快速求连续区间和，我们可以使用前缀和
如图

计算完前缀和后

代码

class Solution {
public:
	int preSum[20010];
	void init(const vector<int>& v) {
		memset(preSum, 0, sizeof preSum);

		for (int i = 0; i < v.size(); i++) {
			preSum[i + 1] = preSum[i] + v[i];
		}

		return;
	}

	int subarraySum(vector<int>& nums, int k) {
		init(nums);
		unordered_map<int, vector<int>> mm;
		mm[0].push_back(0);
		int ans = 0;
		for (int i = 1; i <= nums.size(); i++) {
			int find = preSum[i] -k;
			if (mm.count(find) != 0) {
				ans += mm[find].size();
			}
			mm[preSum[i]].push_back(i);
		}

		return ans;
	}
};

class Solution {
public:
	int preSum[20010];
    //该函数计算前缀和
	void init(const vector<int>& v) {
		memset(preSum, 0, sizeof preSum);
		for (int i = 0; i < v.size(); i++) {
			preSum[i + 1] = preSum[i] + v[i];
		}

		return;
	}

	int subarraySum(vector<int>& nums, int k) {
        //初始化前缀和
		init(nums);
        //开启哈希表  哈希表的记录意义是 记录等于key的前缀和有多少个
		unordered_map<int, int> mm;
        //记录 前缀和的首个元素
		mm[0]=1;
		int ans = 0;
        //遍历前缀和  假设sum[x]已经得到了， 我们查找 sum[x]-sum[y] =K中的sum[y]
		for (int i = 1; i <= nums.size(); i++) {
            //将 sum[x]-sum[y] =K方程变换一下 也就是要找sum[y] = sum[x]-k
            //也就是 要找preSum[i]-k;
			int find = preSum[i] -k;
            //直接从哈希表中 有没有记录匹配这个find值的前缀和
			if (mm.count(find) != 0) {
                //如果有 就记录
                //也就是记录 有几个匹配 Presum[i] - find = K
				ans += mm[find];
			}
            //遍历的过程中 遍历完的前缀和元素要记录进哈希表
			mm[preSum[i]]++;
		}

		return ans;
	}
};