其他分享
首页 > 其他分享> > 力扣_560 和为k的子数组

力扣_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

来源:力扣(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;
    }
}

侵权请联系删除。

标签:HashMap,nums,560,sum0,力扣,int,数组,preSum,前缀
来源: https://blog.csdn.net/m0_53328738/article/details/123164986