README.md

560. 和为 K 的子数组

English Version

题目描述

给定一个整数数组和一个整数 k，你需要找到该数组中和为 k 的连续的子数组的个数。

示例 1 :

输入:nums = [1,1,1], k = 2
输出: 2 , [1,1] 与 [1,1] 为两种不同的情况。

说明 :

1. 数组的长度为 [1, 20,000]。
2. 数组中元素的范围是 [-1000, 1000] ，且整数 k 的范围是 [-1e7, 1e7]。

解法

Python3

class Solution:
    def subarraySum(self, nums: List[int], k: int) -> int:
        mp = Counter()
        mp[0] = 1
        res = s = 0
        for num in nums:
            s += num
            res += mp[s - k]
            mp[s] += 1
        return res

Java

class Solution {
    public int subarraySum(int[] nums, int k) {
        Map<Integer, Integer> map = new HashMap<>();
        map.put(0, 1);
        int res = 0;
        int s = 0;
        for (int num : nums) {
            s += num;
            res += map.getOrDefault(s - k, 0);
            map.put(s, map.getOrDefault(s, 0) + 1);
        }
        return res;
    }
}

TypeScript

function subarraySum(nums: number[], k: number): number {
    let ans = 0,
        pre = 0;
    let hashTable = new Map();
    hashTable.set(0, 1);
    for (let num of nums) {
        pre += num;
        ans += hashTable.get(pre - k) || 0;
        hashTable.set(pre, (hashTable.get(pre) || 0) + 1);
    }
    return ans;
}

C++

class Solution {
public:
    int subarraySum(vector<int>& nums, int k) {
        unordered_map<int, int> mp;
        mp[0] = 1;
        int res = 0, s = 0;
        for (int num : nums)
        {
            s += num;
            res += mp[s - k];
            ++mp[s];
        }
        return res;
    }
};

Go

func subarraySum(nums []int, k int) int {
	mp := make(map[int]int)
	mp[0] = 1
	res, s := 0, 0
	for _, num := range nums {
		s += num
		res += mp[s-k]
		mp[s]++
	}
	return res
}

...