C++——和为k的子数组leetcode560

题目描述

给你一个整数数组 nums 和一个整数 k ，请你统计并返回 该数组中和为 k 的子数组的个数 。
子数组是数组中元素的连续非空序列。
示例 1：
输入：nums = [1,1,1], k = 2
输出：2
示例 2：
输入：nums = [1,2,3], k = 3
输出：2

思路

为了减少时间复杂度。我们发现当遍历到nums[i]的时候，如果我们能够知道此时的前i项和sum_i，并且能够以O(1)查询前i-1个数字的前s项和【s=0~(i-1)】，那么当前所需的那个前s项和就等于sum_i-k。这样[s+1到i]就是目标子数组。
因此考虑用一个哈希表在遍历过程中添加前i项和数值以及出现的次数，并在遍历过程中判断。

#include <vector>
#include <unordered_map>
using namespace std;

int solution(vector<int> nums, int k)
{
	int n = nums.size(), sum_i = 0, ans = 0;
	unordered_map<int, int> hash_map;
	hash_map[0] = 1; // 这样可以让子数组从索引0开始，也就是前i项和为0的至少有一个
	for(int i = 0; i<n; i++)
	{
		sum_i += nums[i]; // 前i项和
		if(hash_map.count(sum_i-k)!=0)
			ans += hash_map[sum_i-k];
		hash_map[sum_i]++;
	}
	return ans;
}