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;
    }
}