LeetCode 560 和为K的子数组

记录一下LeetCode 刷题的心路历程，以增加自己的记忆与理解。本人C++小白，大佬勿喷。如果对文内内容可以提出建议或帮助，感激不尽。

560. 给你一个整数数组 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

枚举方法不作赘述，这里提供一种前缀和和unordered_map的实现，可以有效降低复杂度为O（n）

class Solution {

public:

    int subarraySum(vector<int>& nums, int k) {

        unordered_map <int,int> mp;

        mp[0] = 1;

        int pre = 0;

        int count = 0;

        for ( auto & m : nums) {

            pre+=m;

            if(mp.find(pre-k) !=mp.end()){

                count+=mp[pre-k];

            }

            mp[pre]++;

        }

        return count;

    }

};

降低复杂度实现的方法：利用前缀表避免反复计算连续数组和，而unordered_map 对键的查找时间复杂度为O（1）