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

发布时间:2023年12月17日

题目描述:

给你一个整数数组?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;
    }
};

文章来源:https://blog.csdn.net/m0_73865858/article/details/134953416
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。