力扣560.和为K的子数组

最新推荐文章于 2025-05-21 15:08:20 发布

liuyang-neu

最新推荐文章于 2025-05-21 15:08:20 发布

阅读量241

点赞数 3

分类专栏：前缀和及其扩展文章标签： leetcode 算法 java

本文链接：https://blog.csdn.net/qq_51352130/article/details/148063375

版权

前缀和及其扩展专栏收录该内容

5 篇文章

订阅专栏

文章目录

题目介绍
题解

题目介绍

在这里插入图片描述

题解

前缀和+哈希表（两数之和）：
在这里插入图片描述

代码如下：

class Solution {
    public int subarraySum(int[] nums, int k) {
        int n = nums.length;
        int[] s = new int[n + 1];
        for (int i = 0; i < n; i++) {
            s[i + 1] = s[i] + nums[i];
        }

        int res = 0;
        Map<Integer, Integer> map = new HashMap<>(n + 1); // 设置容量可以快 2ms
        for (int he : s) {
            res += map.getOrDefault(he - k, 0);
            map.put(he, map.getOrDefault(he, 0) + 1); // map[he]++
        }
        return res;
    }
}

错解：这种方法只适用于数组中所有元素都是正数的情况。当数组中包含负数时，滑动窗口就不能保证正确性了。例如，nums = [-1, -1, 1] 和 k = 0，你的算法会漏掉 [-1, -1, 1] 这个子数组。

class Solution {
    public int subarraySum(int[] nums, int k) {
        int res = 0, len = nums.length, l = 0, sum = 0;
        for (int r = 0; r < len; r++) {
            sum += nums[r];
            if (sum == k) {
                res++;
            }
            while (sum > k && l <= r) {
                sum -= nums[l++];
                if (sum == k) {
                    res++;
                }
            }
        }
        return res;
    }
}