838. Subarray Sum Equals K

最新推荐文章于 2025-04-22 09:11:01 发布

Vest-Lee

最新推荐文章于 2025-04-22 09:11:01 发布

阅读量177

点赞数

CC 4.0 BY-SA版权

分类专栏：刷题

本文链接：https://blog.youkuaiyun.com/vestlee/article/details/80720904

刷题专栏收录该内容

215 篇文章

订阅专栏

本文介绍了一种高效算法，用于找出数组中所有连续子数组的总数，这些子数组的和等于给定整数K。通过使用哈希表记录前缀和出现的次数，可以在O(n)的时间复杂度内解决此问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

描述
Given an array of integers and an integer k, you need to find the total number of continuous subarrays whose sum equals to k.

您在真实的面试中是否遇到过这个题？  是

样例
Input:nums = [1,1,1], k = 2
Output: 2

基本思路都差不多，就是建立累加数组。使用关联列表会更简单一点。

class Solution {
public:
    /**
     * @param nums: a list of integer
     * @param k: an integer
     * @return: return an integer, denote the number of continuous subarrays whose sum equals to k
     */
    int subarraySumEqualsK(vector<int> &nums, int k) {
        // write your code here
        int res = 0, sum = 0, n = nums.size();
        unordered_map<int, int> m{{0, 1}};
        for (int i = 0; i < n; ++i) {
            sum += nums[i];
            res += m[sum - k];
            ++m[sum]; 
        }
        return res;
    }
};