題目描述

給定一個整數數組和一個整數 k，你需要找到該數組中和為 k 的連續的子數組的個數。

示例 1 :

輸入:nums = [1,1,1], k = 2
輸出: 2 , [1,1] 與 [1,1] 為兩種不同的情況。

說明 :

1. 數組的長度為 [1, 20,000]。
2. 數組中元素的範圍是 [-1000, 1000] ，且整數 k 的範圍是 [-1e7, 1e7]。

解題思路

維護一個map，在遍歷數組時，更新包含當前數字之前所有數的和出現的次數，這樣每遍歷到一個位置，將當前和減去k，若map中出現了此和，則說明一定會有一個連續的子數組和為k，所以使結果加上此和出現的次數。

代碼

 1 class Solution {
 2 public:
 3     int subarraySum(vector<int>& nums, int k) {
 4         int res = 0, sum = 0;
 5         map<int, int> kmap;
 6         kmap[0] = 1;
 7         for(int num: nums){
 8             sum += num;
 9             res += kmap[sum - k];
 
10             kmap[sum]++;
11         }
12         return res;
13     }
14 };

LeetCode 560. 和為K的子數組（Subarray Sum Equals K）