leetcode-hot100-前缀和-优快云博客

本文链接：https://blog.youkuaiyun.com/qq_41784433/article/details/121809073

本文介绍了如何利用前缀和解决LeetCode第560题——和为K的子数组问题。提供了两种解决方案，一种是枚举法，另一种是使用HashMap记录前缀和并统计结果。通过这两种方法，可以有效地找到数组中和为K的子数组个数。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

文章目录

[560. 和为K的子数组 - 力扣（LeetCode） (leetcode-cn.com)](https://leetcode-cn.com/problems/subarray-sum-equals-k/)

560. 和为K的子数组 - 力扣（LeetCode） (leetcode-cn.com)

思路一：进行枚举

考虑以i结尾和正好为k的连续子数组的个数，就需要统计符合条件的下标j的个数，j∈[0,i]并且[j…i]这个子数组的和正好为k。
枚举[0…i]里所有的下标j来判断是否符合条件。

代码

public class Solution {
    public int subarraySum(int[] nums, int k) {
        int count = 0;
        //从前到后，再后到前进行求和计算。
        for (int start = 0; start < nums.length; ++start) {
            int sum = 0;
            for (int end = start; end >= 0; --end) {
                sum += nums[end];
                if (sum == k) {
                    count++;
                }
            }
        }
        return count;
    }
}

思路二：前缀和：

使用HashMap。key为每次遍历的前缀和，value是前缀和对应出现的次数。
最开的时候先put(0,1)表示前缀和为0的只有一个。
遍历的时候求和，然后看这个map中是否包含preSum - k这个key，有就相加。
不管是否有，都将每次的前缀和放入进map中。

代码：

class Solution {
    public int subarraySum(int[] nums, int k) {
        int n = nums.length;
        //前缀和+HashMap
        Map<Integer, Integer> preMap = new HashMap<>();
        //先默认只有一个数能凑齐0
        preMap.put(0,1);
        int res = 0;
        int preSum = 0;
        for(int num : nums){
            preSum += num;
            if(preMap.containsKey(preSum - k)){
                res += preMap.get(preSum - k);
            }
            preMap.put(preSum, preMap.getOrDefault(preSum, 0) + 1);

        }
        return res;

    }
}