560. Leetcode 和为k的子数组
题目链接: 560. 和为K的子数组
题目描述
给定一个整数数组和一个整数 **k,**你需要找到该数组中和为 k 的连续的子数组的个数。
样例
示例1:
输入:nums = [1,1,1], k = 2
输出: 2 , [1,1] 与 [1,1] 为两种不同的情况。
说明:
- 数组的长度为 [1, 20,000]。
- 数组中元素的范围是 [-1000, 1000] ,且整数 k 的范围是 [-1e7, 1e7]。
ws
算法1 (前缀和 + 暴力枚举)
思路分析
数组长度为 n, n的数量级为 1 0 4 10^4 104, 虽然用前缀和预处理了前缀和数组, 但最后枚举所有的区间情况这步的复杂度为 O ( n 2 ) O(n^2) O(n2),故这一步会超时.
时间复杂度 O ( n 2 ) O(n^2) O(n2)
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 ) O(n) 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;
}
};