Java版LeetCode

Appearance

LCR 010. 和为 K 的子数组 ​

题目 ​

给定一个整数数组和一个整数 k， 请找到该数组中和为 k的连续子数组的个数。

示例 1：

输入:nums = [1,1,1], k = 2
输出: 2
解释: 此题 [1,1] 与 [1,1] 为两种不同的情况
输入:nums = [1,1,1], k = 2
输出: 2
解释: 此题 [1,1] 与 [1,1] 为两种不同的情况

示例 2：

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

提示:

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

注意：本题与主站 560 题相同： https://leetcode-cn.com/problems/subarray-sum-equals-k/

题解 ​

java
public int subarraySum(int[] nums, int k) {
    int result = 0, sum = 0;
    Map<Integer, Integer> prefix = new HashMap<>();
    // 前缀和为0的个数 仅作为初始化
    prefix.put(0, 1);

    for (int num : nums) {
        // 前缀和累加
        sum += num;
        // 若存在前缀和为sum-k则为解
        result += prefix.getOrDefault(sum - k, 0);
        // 合并前缀和为sum的子数组数量
        prefix.merge(sum, 1, Integer::sum);
    }

    return result;
}
public int subarraySum(int[] nums, int k) {
    int result = 0, sum = 0;
    Map<Integer, Integer> prefix = new HashMap<>();
    // 前缀和为0的个数 仅作为初始化
    prefix.put(0, 1);

    for (int num : nums) {
        // 前缀和累加
        sum += num;
        // 若存在前缀和为sum-k则为解
        result += prefix.getOrDefault(sum - k, 0);
        // 合并前缀和为sum的子数组数量
        prefix.merge(sum, 1, Integer::sum);
    }

    return result;
}