【数据结构-哈希&&前缀】力扣560. 和为 K 的子数组

最新推荐文章于 2024-11-26 21:30:34 发布

hlc@

最新推荐文章于 2024-11-26 21:30:34 发布

阅读量212

点赞数 3

分类专栏：数据结构文章标签：数据结构哈希算法 leetcode

本文链接：https://blog.csdn.net/sjsjs11/article/details/140792013

版权

数据结构专栏收录该内容

94 篇文章 0 订阅

订阅专栏

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

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

示例 1：
输入：nums = [1,1,1], k = 2
输出：2

示例 2：
输入：nums = [1,2,3], k = 3
输出：2

在这里插入图片描述

class Solution {
public:
    int subarraySum(vector<int>& nums, int k) {
        unordered_map<int, int> group;
        int sum = 0;
        int res = 0;
        group[0] = 1;
        for(auto &num : nums){
            sum += num;
            res += group[sum - k];
            group[sum]++;
        }
        return res;
    }
};