Leetcode 560. 和为 K 的子数组 前缀和 哈希
地址 https://leetcode-cn.com/problems/subarray-sum-equals-k/
给你一个整数数组 nums 和一个整数 k ,请你统计并返回该数组中和为 k 的连续子数组的个数。
示例 1:
输入:nums = [1,1,1], k = 2
输出:2
示例 2:
输入:nums = [1,2,3], k = 3
输出:2
提示:
1 <= nums.length <= 2 * 10^4
-1000 <= nums[i] <= 1000
-10^7 <= k <= 10^7
解答
快速求连续区间和,我们可以使用前缀和
如图
计算完前缀和后
代码
class Solution {
public:
int preSum[20010];
void init(const vector<int>& v) {
memset(preSum, 0, sizeof preSum);
for (int i = 0; i < v.size(); i++) {
preSum[i + 1] = preSum[i] + v[i];
}
return;
}
int subarraySum(vector<int>& nums, int k) {
init(nums);
unordered_map<int, vector<int>> mm;
mm[0].push_back(0);
int ans = 0;
for (int i = 1; i <= nums.size(); i++) {
int find = preSum[i] -k;
if (mm.count(find) != 0) {
ans += mm[find].size();
}
mm[preSum[i]].push_back(i);
}
return ans;
}
};
class Solution {
public:
int preSum[20010];
//该函数计算前缀和
void init(const vector<int>& v) {
memset(preSum, 0, sizeof preSum);
for (int i = 0; i < v.size(); i++) {
preSum[i + 1] = preSum[i] + v[i];
}
return;
}
int subarraySum(vector<int>& nums, int k) {
//初始化前缀和
init(nums);
//开启哈希表 哈希表的记录意义是 记录等于key的前缀和有多少个
unordered_map<int, int> mm;
//记录 前缀和的首个元素
mm[0]=1;
int ans = 0;
//遍历前缀和 假设sum[x]已经得到了, 我们查找 sum[x]-sum[y] =K中的sum[y]
for (int i = 1; i <= nums.size(); i++) {
//将 sum[x]-sum[y] =K方程变换一下 也就是要找sum[y] = sum[x]-k
//也就是 要找preSum[i]-k;
int find = preSum[i] -k;
//直接从哈希表中 有没有记录匹配这个find值的前缀和
if (mm.count(find) != 0) {
//如果有 就记录
//也就是记录 有几个匹配 Presum[i] - find = K
ans += mm[find];
}
//遍历的过程中 遍历完的前缀和元素要记录进哈希表
mm[preSum[i]]++;
}
return ans;
}
};
作 者: itdef
欢迎转帖 请保持文本完整并注明出处
技术博客 http://www.cnblogs.com/itdef/
B站算法视频题解
https://space.bilibili.com/18508846
qq 151435887
gitee https://gitee.com/def/
欢迎c c++ 算法爱好者 windows驱动爱好者 服务器程序员沟通交流
如果觉得不错,欢迎点赞,你的鼓励就是我的动力
欢迎转帖 请保持文本完整并注明出处
技术博客 http://www.cnblogs.com/itdef/
B站算法视频题解
https://space.bilibili.com/18508846
qq 151435887
gitee https://gitee.com/def/
欢迎c c++ 算法爱好者 windows驱动爱好者 服务器程序员沟通交流
如果觉得不错,欢迎点赞,你的鼓励就是我的动力
![阿里打赏](https://files.cnblogs.com/files/itdef/ali2.bmp)
![微信打赏](https://files.cnblogs.com/files/itdef/weixin2.bmp)