ARTICLE DETAIL

建站实战干货

来自一线的建站与推广经验沉淀,每一条都经过真实交付验证。

每日一题:Leetcode560.和为K的子数组(前缀和+哈希)

2026/8/14 4:06:26 拓冰建站 浏览量
每日一题:Leetcode560.和为K的子数组(前缀和+哈希)

题目描述:

给你一个整数数组 nums 和一个整数 k ,请你统计并返回 该数组中和为 k 的子数组的个数 

子数组是数组中元素的连续非空序列。

示例 1:

输入:nums = [1,1,1], k = 2
输出:2

示例 2:

输入:nums = [1,2,3], k = 3
输出:2

提示:

  • 1 <= nums.length <= 2 * 104
  • -1000 <= nums[i] <= 1000
  • -107 <= k <= 107

思路:遍历数组使用一个变量presum记录当前节点的前缀和,用另一个指针去遍历该位置前面的数组,找到和为k的子数组,但是这样的话时间复杂度就是o(n^2)

优化:采用哈希表记录前缀和,把找和为k的子数组转化为在哈希表中找和为presum-k的前缀和数组,如果哈希表中存在则将结果ret加上该前缀和所对应的哈希值(即出现的次数),再将该位置的前缀和放入哈希表中。注意特殊情况,如果presum==k则将ret++。

代码实现:

class Solution {
public:int subarraySum(vector<int>& nums, int k) {int presum=0,ret=0;unordered_map<int,int> hash;for(auto e: nums){presum+=e;if(hash.count(presum-k))ret+=hash[presum-k];if(presum==k)ret++;hash[presum]++;}return ret;}
};