Leetcode 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

解题思路

本题是找出所有中和为 k 的子数组的个数，子数组的和可以用前缀和来表示，因此是一个前缀和的问题

代码实现

class Solution {public int subarraySum(int[] nums, int k) {int res = 0;int[] preSum = new int[nums.length];preSum[0] = nums[0];for (int i = 1; i < nums.length; i++) {preSum[i] = preSum[i -1] +nums[i];}for (int i = 0; i < preSum.length; i++) {if (preSum[i] == k) {res += 1;}for (int j = 0; j < i; j++) {if (preSum[i] - preSum[j] == k) {res += 1;}}}return res;}
}