和为 K 的子数组 C++题解

和为 K 的子数组

内存限制： 256 MiB 时间限制： 50 ms 标准输入输出 题目类型： 传统 评测方式： 文本比较

题目描述

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

输入格式

第一行输入 n 和 k

第二行输入 n 个整数，表示数组 nums

输出格式

输出一行，包含一个数字，表示和为 k 的连续子数组的个数

样例

样例输入 1

复制3 2
1 1 1

样例输出 1

复制2

样例输入 2

复制3 3
1 2 3

样例输出 2

复制2

样例输入 3

复制10 3
1 1 1 -3 3 -3 3 -1 2 1

样例输出 3

复制9

数据范围与提示

1 <= nums.length <= 2 * 10^4

-1000 <= nums[i] <= 1000

-10^7 <= k <= 10^7

#include <bits/stdc++.h>
using namespace std;
const int maxn = 6e7 + 5;
int a, n, k, ans;
int sum[20005], cnt[maxn]; 
int main() {
	scanf("%d %d", &n, &k);
	for(int i = 1; i <= n; i++) {
		scanf("%d", &a);
		sum[i] = sum[i - 1] + a;
	} 
	cnt[30000000] = 1;
	for(int i = 1; i <= n; i++) {
		ans += cnt[sum[i] - k + 30000000];
		cnt[sum[i] + 30000000]++;
	} 
	printf("%d", ans);
	return 0;
}