LeetCode 560 Subarray Sum Equals K【前綴和】

傳送門

題意: 有多少個連續子序列和爲給定的k,

思路: 連續, 就要想到前綴和, 所以我們利用前綴和的性質, 相減等於中間的和, 那麼直接記錄每個前綴的次數, 然後累加即可, 注意這類問題 0 這個數都是會作爲默認點加的! 不要忘了，想想也是對的

AC Code

class Solution {
public:
    int subarraySum(vector<int>& nums, int k) {
        map<int, int>mp; mp[0] = 1;
        int sum = 0, ans = 0;
        for (int i = 0 ; i < nums.size() ; ++ i) {
            sum += nums[i];
            ans += mp[sum-k];
            ++ mp[sum];
        }
        return ans;
    }
};