Leetcode 560.和为k的子数组

560. Leetcode 和为k的子数组

题目链接: 560. 和为K的子数组

题目描述

给定一个整数数组和一个整数 **k，**你需要找到该数组中和为 k 的连续的子数组的个数。

样例

示例1: 
输入:nums = [1,1,1], k = 2
输出: 2 , [1,1] 与 [1,1] 为两种不同的情况。

---

说明:

1. 数组的长度为 [1, 20,000]。
2. 数组中元素的范围是 [-1000, 1000] ，且整数 k 的范围是 [-1e7, 1e7]。
   ws

算法1 (前缀和 + 暴力枚举)

思路分析

数组长度为 n, n的数量级为 $10^4$, 虽然用前缀和预处理了前缀和数组, 但最后枚举所有的区间情况这步的复杂度为 $O(n^2)$,故这一步会超时.

时间复杂度 $O(n^2)$

C++ 代码

class Solution 
{
public:
    static constexpr int N = 2 * 1e+4 + 10;
    int s[N]; 
    int subarraySum(vector<int>& nums, int k) 
    {
        s[0] = 0; 
        int res = 0, n = nums.size();
        for(int i = 1; i <= n; ++i) s[i] = s[i - 1] + nums[i - 1];  // 预处理前缀和数组

        for(int i = 1; i <= n; ++i)                                // 区间左端点
            for(int j = 1; j <= i; ++j)                            // 区间右端点
                if(s[i] - s[j - 1] == k) ++res; 
     
        return res; 
    }

};

---

算法2 (前缀和 + 哈希表)

思路分析

经过前缀和预处理得到前缀和数组s, 可以得到一段区间的和[j , i] 为 s[i] - s[j - 1]. 由于我们找的是区间和为k的一段区间, 故满足 s[i] - s[j - 1] = k. 即 s[i] - k = s[j - 1]. 故我们枚举以下标为i右端点的区间时, 我们可以找一下其前面有没有存在以j为右端点区间, 且这段区间和为为 s[i] - k. 故快速找到这样一个区间, 我们可以用一个哈希表来存储。存储当前下标i,其前面有多少个s[1 ~ i - 1] 的值为 s[i] - k. 这样我们只需遍历一遍即可找出所有区间和为k的情况.

时间复杂度 $O(n)$

C++ 代码

class Solution 
{
public:
    static constexpr int N = 2 * 1e+4 + 10;
    int s[N]; 
    int subarraySum(vector<int>& nums, int k) 
    {
        s[0] = 0; 
        int res = 0, n = nums.size();
        for(int i = 1; i <= n; ++i) s[i] = s[i - 1] + nums[i - 1];  // 预处理前缀和
         
        unordered_map<int, int> m;
        m[0] = 1;                                     // s[0]为0, 故先+1; 
        
        for(int i = 1; i <= n; ++i)                                  
        {
            res += m[s[i] - k];                      
            m[s[i]]++; 
        }
        return res; 
    }
};

---