每日一题: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;}
};