【LeetCode】热题100：和为k的子数组

最新推荐文章于 2024-05-20 20:08:58 发布

不要爆爆了吗

最新推荐文章于 2024-05-20 20:08:58 发布

阅读量153

点赞数 3

文章标签：算法数据结构 leetcode java

本文链接：https://blog.csdn.net/Snake_____/article/details/136573049

版权

题目：

给你一个整数数组 nums 和一个整数 k ，请你统计并返回该数组中和为 k 的子数组的个数。
子数组是数组中元素的连续非空序列。

示例 1：
输入：nums = [1,1,1], k = 2
输出：2

示例 2：
输入：nums = [1,2,3], k = 3
输出：2

提示：
1 <= nums.length <= 2 * 10的4次方
-1000 <= nums[i] <= 1000
-10的7次方 <= k <= 10的7次方

解题思路：使用哈希表来存储前缀和及其出现的次数，初始化哈希表，将前缀和为0的出现次数设为1。遍历数组，计算当前位置的前缀和，并在哈希表中查找是否存在前缀和减去 k 的值，若存在，则表示当前位置到前面某个位置的子数组和为 k，累加相应计数，更新当前前缀和的出现次数。最后返回累加的计数值，即为和为 k 的子数组数量。

class Solution {
    public int subarraySum(int[] nums, int k) {
        // 使用 HashMap 存储前缀和及其出现次数
        HashMap<Integer, Integer> prefixSumCount = new HashMap<>();
        prefixSumCount.put(0, 1); // 初始化，前缀和为0出现1次
        int count = 0;
        int prefixSum = 0;

        // 遍历数组，计算前缀和，并统计和为 k 的子数组数量
        for (int num : nums) {
            prefixSum += num;
            // 如果前缀和减去 k 在 HashMap 中存在，则表示当前位置与该位置的差值为 k，因此计数加上该前缀和的出现次数
            if (prefixSumCount.containsKey(prefixSum - k)) {
                count += prefixSumCount.get(prefixSum - k);
            }
            // 更新当前前缀和的出现次数
            prefixSumCount.put(prefixSum, prefixSumCount.getOrDefault(prefixSum, 0) + 1);
        }

        return count;
    }
}