1. 程式人生 > 其它 >leetcode560. 和為 K 的子陣列(字首和+hash)

leetcode560. 和為 K 的子陣列(字首和+hash)

連結:https://leetcode-cn.com/problems/subarray-sum-equals-k/

題目

  1. 和為 K 的子陣列
    給你一個整數陣列 nums 和一個整數 k ,請你統計並返回該陣列中和為 k 的連續子陣列的個數。

用例

示例 1:

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

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

提示:

1 <= nums.length <= 2 * 104
-1000 <= nums[i] <= 1000
-107 <= k <= 107

思路

做這題一開始沒有注意到陣列中存在負數 按照揹包的思想去做
但實際上這是一道標準的字首和問題
對於”xxxx個連續元素的值達成xx條件的問題“都可以用字首和完成

注意初始化hash表
mp[0]=1 當前綴和為0(從頭開始)的情況
遍歷陣列
記錄累加字首和值pre
查詢pre-target的值在hash表中是否存在相同的字首和 則增加其計數
把當前pre加入mp

class Solution {
public:
    int subarraySum(vector<int>& nums, int k) {
        unordered_map<int,int>mp;
        mp[0]=1;
        int count=0,pre=0;
        for(auto &x:nums){
            pre+=x;
            if(mp.find(pre-k)!=mp.end()){
                count+=mp[pre-k];
            }
            mp[pre]++;
        }
        return count;
    }
};