560. 和为 K 的子数组-哈希表

一、题目描述

给你一个整数数组 nums 和一个整数 k ，请你统计并返回 该数组中和为 k 的子数组的个数 。

示例 1：

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

示例 2：

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

二、解题

哈希表

class Solution {
    public int subarraySum(int[] nums, int k) {
        //哈希表实现
        int cnt = 0,sum = 0;
        int length = nums.length;
        Map<Integer,Integer> map = new HashMap<>();
        map.put(0,1);
        for(int i = 0;i<length;i++){
            sum += nums[i]; 
            //如果哈希表中存在之前的值 则取值
            if(map.containsKey(sum - k)){
                cnt += map.get(sum - k);
            }
            map.put(sum,map.getOrDefault(sum,0)+1);
        }
        return cnt;
    }
}

时间复杂度：O(n)；

空间复杂度：O(n)。