【LeetCode 前缀和 + 哈希表】LC_560_和为K的子数组

文章目录

      • [1. 和为K的子数组🆗](#1. 和为K的子数组🆗)

1. 和为K的子数组🆗

题目链接🔗


  • 🐧解题思路前缀和 + 哈希表

🍎 设i为数组中的任意位置,⽤ sum[i] 表⽰ [0, i] 区间内所有元素的和。

🍎 想知道有多少个「以 i 为结尾的和为 k 的⼦数组」,就要找到有多少个起始位置为 x1, x2, x3... 使得 [x, i] 区间内的所有元素的和为 k 。那么[0, x]区间内的和是不是就是sum[i] - k 了。

于是问题就变成:

找到在[0, i - 1]区间内,有多少前缀和等于sum[i] - k的即可。

我们不⽤真的初始化⼀个前缀和数组,因为我们只关⼼在i位置之前,有多少个前缀和等于
sum[i] - k 。因此,我们仅需⽤⼀个哈希表,⼀边求当前位置的前缀和,⼀边存下之前每⼀种

前缀和出现的次数。


cpp 复制代码
class Solution
{
public:
 int subarraySum(vector<int>& nums, int k) 
 {
	 unordered_map<int, int> hash; // 统计前缀和出现的次数
	 
	 // 为什么初始值 hash[0] = 1呢?
	 // 因为:有可能 sum == k, 我们统计 hash.count(sum - k)统计的是当前元素之前的值,没有统计千
	 // 前缀和刚好等于 k 的情况
	 hash[0] = 1;
	 
	 int sum = 0, ret = 0;
	 for(auto x : nums)
	 {
		 sum += x; // 计算当前位置的前缀和
		 if(hash.count(sum - k)) ret += hash[sum - k]; // 统计个数
		 hash[sum]++;
	 }
	 return ret;
 }
};
相关推荐
Fanxt_Ja1 天前
【LeetCode】算法详解#15 ---环形链表II
数据结构·算法·leetcode·链表
元亓亓亓1 天前
LeetCode热题100--105. 从前序与中序遍历序列构造二叉树--中等
算法·leetcode·职场和发展
仙俊红2 天前
LeetCode每日一题,20250914
算法·leetcode·职场和发展
_不会dp不改名_2 天前
leetcode_21 合并两个有序链表
算法·leetcode·链表
吃着火锅x唱着歌2 天前
LeetCode 3302.字典序最小的合法序列
leetcode
睡不醒的kun2 天前
leetcode算法刷题的第三十四天
数据结构·c++·算法·leetcode·职场和发展·贪心算法·动态规划
吃着火锅x唱着歌2 天前
LeetCode 978.最长湍流子数组
数据结构·算法·leetcode
爱编程的化学家2 天前
代码随想录算法训练营第十一天--二叉树2 || 226.翻转二叉树 / 101.对称二叉树 / 104.二叉树的最大深度 / 111.二叉树的最小深度
数据结构·c++·算法·leetcode·二叉树·代码随想录
吃着火锅x唱着歌2 天前
LeetCode 1446.连续字符
算法·leetcode·职场和发展
愚润求学2 天前
【贪心算法】day10
c++·算法·leetcode·贪心算法