力扣_560 和为k的子数组

~北木南

已于 2022-10-26 15:01:45 修改

阅读量82

点赞数

分类专栏：力扣文章标签： leetcode 散列表蓝桥杯

于 2022-02-27 15:47:20 首次发布

本文链接：https://blog.csdn.net/m0_53328738/article/details/123164986

版权

力扣专栏收录该内容

5 篇文章 0 订阅

订阅专栏

给你一个整数数组 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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/subarray-sum-equals-k
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

先来看一下HashMap的定义：

散列表（Hash table，也叫哈希表），是根据关键码值(Key value)而直接进行访问的数据结构。也就是说，它通过把关键码值映射到表中一个位置来访问记录，以加快查找的速度。这个映射函数叫做散列函数，存放记录的数组叫做散列表。

HashMap的存储结构：

HashMap是采用数组+链表/红黑树的方式来存储数据

在这里插入图片描述

简单来说就类似于：Python的字典结构。

接下来说一下解题中的HashMap：

//创建一个哈希表
HashMap<Integer, Integer> preSum = new HashMap<>();

//向哈希表中增加一个元素。键为0，value为1
preSum.put(0, 1);

//判断哈希表中是否存在键为1的元素，返回Boolean类型
preSum.containsKey(1)

//获得键为2的元素对应的值，不存在该键则返回自定义的值0
preSum.getOrDefault(2, 0);

解题代码：

class Solution {
    public int subarraySum(int[] nums, int k) {
        int n = nums.length;
        // map：前缀和 -> 该前缀和出现的次数
        HashMap<Integer, Integer> 
            preSum = new HashMap<>();
        // base case
        preSum.put(0, 1);

        int res = 0, sum0_i = 0;
        for (int i = 0; i < n; i++) {
            sum0_i += nums[i];
            // 这是我们想找的前缀和 nums[0..j]
            int sum0_j = sum0_i - k;
            // 如果前面有这个前缀和，则直接更新答案
            if (preSum.containsKey(sum0_j))
                res += preSum.get(sum0_j);
            // 把前缀和 nums[0..i] 加入并记录出现次数
            preSum.put(sum0_i, 
                preSum.getOrDefault(sum0_i, 0) + 1);
        }
        return res;
    }
}

侵权请联系删除。