560. 和为 K 的子数组

原题链接：560. 和为 K 的子数组

 

solution:

枚举：题目求解的是和为K的连续子数组，因此从每个区间终点开始向前枚举即可。

时间复杂度O(n^2)

class Solution {
public:
    int subarraySum(vector<int>& nums, int k) {
        int n = nums.size();
        int res = 0;    //定义返回值
        for(int i = 0;i < n;i++) {
            int sum = 0;
            for(int j = i;j >= 0;j--) {
                sum += nums[j];
                if(sum == k) res++;
            }
        }
        return res;
    }
};

hash+prefix优化

因为求解的连续区间和为K的情况，相当于求两个前缀和差为k的情况因此可以用前缀和和哈希表优化。

时间复杂度O(n^2)

class Solution {
public:
    int subarraySum(vector<int>& nums, int k) {
        int n = nums.size();
        unordered_map<int,int> map; //保存前缀和
        int res = 0;    //定义返回值
        vector<int> prefix(n + 1);
        for(int i = 1;i <= n;i++) {
            prefix[i] = prefix[i - 1] + nums[i - 1];
        }
        map[0] = 1;    //处理前缀和刚好为k的情况
        for(int i = 1;i <= n;i++) {
            if(map.count(prefix[i] - k)) res += map[prefix[i] - k];
            map[prefix[i]]++;
        }
       
        return res;
    }
};