560. 和为 K 的子数组

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

示例 1:

输入:nums = [1,1,1], k = 2
输出:2
示例 2:

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

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

import java.util.HashMap;
import java.util.Map;

class Solution {
    public int subarraySum(int[] nums, int k) {
        if (nums == null || nums.length == 0) {
            return 0;
        }
        Map<Integer, Integer> timesMap = new HashMap<>();
        timesMap.put(0, 1);
        int sum = 0;
        int ret = 0;
        for (int num : nums) {
            sum += num;
            ret += timesMap.getOrDefault(sum - k, 0);
            timesMap.put(sum, timesMap.getOrDefault(sum, 0) + 1);
        }
        return ret;
    }
}
posted @ 2021-12-23 22:33  Tianyiya  阅读(27)  评论(0)    收藏  举报