diff --git a/子串/1. 和为K的子数组.md b/子串/1. 和为K的子数组.md new file mode 100644 index 0000000..38d6f1f --- /dev/null +++ b/子串/1. 和为K的子数组.md @@ -0,0 +1,59 @@ +# 和为 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 + + + +## 思路 + +- 前缀和 + - `prefixSum` +- 哈希表单次遍历 + +## 代码 + +```java +class Solution { + public int subarraySum(int[] nums, int k) { + // Init + // Prefix Sum + Map map = new HashMap<>(); + map.put(0, 1); + int ans = 0, pre = 0; + + // Traverse + for (int i = 0; i < nums.length; i++) { + pre += nums[i]; + // Judge + if (map.containsKey(pre - k)) { + ans += map.get(pre - k); + } + map.put(pre, map.getOrDefault(pre, 0) + 1); + } + + return ans; + } +} +``` \ No newline at end of file