【LeetCode: 560. 和为 K 的子数组 + 前缀和 + 哈希表】

🌲 算法刷题专栏 | 面试必备算法 | 面试高频算法 🍀
🌲 越难的东西,越要努力坚持，因为它具有很高的价值，算法就是这样✨
🌲 作者简介：硕风和炜，CSDN-Java领域优质创作者🏆，保研|国家奖学金|高中学习JAVA|大学完善JAVA开发技术栈|面试刷题|面经八股文|经验分享|好用的网站工具分享💎💎💎
🌲 恭喜你发现一枚宝藏博主,赶快收入囊中吧🌻
🌲 人生如棋，我愿为卒，行动虽慢，可谁曾见我后退一步？🎯🎯

🚩 题目链接

• 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

🌟 求解思路&实现代码&运行结果

⚡ 前缀和 + 哈希表

🥦 求解思路

1. 前缀和：通过计算前缀和数组 arr，其中 arr[i] 表示从数组开头到第 i 个元素的和。
2. 哈希表优化：使用哈希表 map 记录前缀和出现的次数。通过检查 arr[i] - k 是否在哈希表中，快速判断是否存在满足条件的子数组。
3. 初始化：哈希表中预先存入 (0, 1)，表示前缀和为 0 的情况出现了一次（即空数组的情况）。
4. 遍历数组：遍历前缀和数组，累加满足条件的子数组数量，并更新哈希表。
5. 有了基本的思路，接下来我们就来通过代码来实现一下。

🥦 实现代码

class Solution {
    public int subarraySum(int[] nums, int k) {
        int n = nums.length;
        int[] arr = new int[n];
        arr[0] = nums[0];
        for (int i = 1; i < n; i++) {
            arr[i] = arr[i - 1] + nums[i];
        }
        HashMap<Integer, Integer> map = new HashMap<>();
        map.put(0, 1);
        int cnt = 0;
        for (int i = 0; i < n; i++) {
            cnt += map.getOrDefault(arr[i] - k, 0);
            map.put(arr[i], map.getOrDefault(arr[i], 0) + 1);
        }
        return cnt;
    }
}

🥦 运行结果

💬 共勉