LeetCode和为k的字数组(560题)

题目展示

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

子数组是数组中元素的连续非空序列。

示例 1：

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

示例 2：

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

题目地址:560. 和为 K 的子数组 - 力扣（LeetCode）

相关类型题目(进阶):2588. 统计美丽子数组数目 - 力扣（LeetCode）

解题思路

首先我们需要搞懂的一个点: 一组数据的和为10 那么我们的目标和也刚好为10 那么这组数据就是满足题目的子数组.

然后我们在延展一下:一组数据和为20 目标和为15 但是这组数据中前二个元素和为5 即我们这组数据将前二个数据剔除,剩下的即为满足的子数组.

代码演示:

class Solution {
    public int subarraySum(int[] nums, int k) {

        int count = 0; // 用来统计和为k的子数组总数
        int sum = 0; // 前缀和

        Map<Integer, Integer> map = new HashMap<Integer, Integer>(nums.length + 1); // 用来统计前缀和的集合

        map.put(0, 1); // key:前缀和 值:出现次数 map.merge(sum,1,Integer::sum); 这行代码会让他变增

        for (int num : nums) {
            sum += num; //计算前缀和
            count += map.getOrDefault(sum - k, 0); 
// 难点:map.getOrDefault(sum - k, DefaultValue) 
//map中getOrDefault方法作用为: 如果有sum-k键返回键值否则返回DefaultValue

            map.merge(sum, 1, Integer::sum); 
// map中的merge 存在sum键 在原值基础上+1 否则 sum键值为1
        }

        return count;
    }
}