Leedcode 560.和为k的子数组 详细注释

560. 和为 K 的子数组

题目：

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

示例 1：

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

示例 2：

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

提示：

• 1 <= nums.length <= 2 * 104
• -1000 <= nums[i] <= 1000
• -107 <= k <= 107

(1)暴力解法（时间复杂度为o(n的平方))

知识点：枚举

#include<iostream>
using namespace std;
int n, sum, cnt;
int k = 7;
int num[100];
int main() {
	cin >> n;
	for (int i = 0; i < n; i++)
	{
		cin >> num[i];
	}
	for (int i = 0; i < n; i++) {
		sum = 0;
		for (int j = i; j < n; j++) {
			sum = sum + num[j];
			cout << "sum:" << sum << "  ,  ";
			if (sum == k) {
				
				cnt++;
				cout << "cnt:" << cnt << endl;
				

			}
		}

	}




	cout << cnt << endl;
	return 0;

}

(2)前缀和+哈希表优化（时间复杂度为o(n))

思路：将每次得到的前缀和保存到哈希数组中，并初始化依次表示初始为一次。例如16-7=9，则证明前缀和为9的数组段到前缀和为16的数组段中一定存在一个数组段和为7的数组,

#include<iostream>
#include<unordered_map>
using namespace std;
int num[100];
int pre;
int n, k=7;
int cnt;
int main() {
	//创建哈希表
	unordered_map<int, int>hash_map;
	hash_map[0] = 1;
	cin >> n;
	for (int i = 0; i < n; i++) {
		cin >> num[i];
	}
	for (int i = 0; i < n; i++) {
		pre = pre + num[i];
		if (hash_map.find(pre - k) != hash_map.end()) {//hash_map.find()作用查找以pre-k为key值 所对应的键值value是否存在  如果不存在返回hash_map.end();
			cnt += hash_map[pre - k];//如果前缀和中出现两个9，cnt++只能加一次 ，而我们因应该加此时的hash_map[pre]=2;

		}
		hash_map[pre]++;//将每次得到的前缀和保存到哈希数组中，并初始化依次表示初始为一次。例如16-7=9，则证明前缀和为9的数组段到前缀和为16的数组段中一定存在一个数组段和为7的数组,



	}

	cout << cnt << endl;

	return 0;
}