560. 和为 K 的子数组

最新推荐文章于 2024-09-13 16:27:15 发布

糖醋_诗酒

最新推荐文章于 2024-09-13 16:27:15 发布

阅读量187

点赞数 8

分类专栏：力扣数据结构训练文章标签：算法 leetcode 数据结构

本文链接：https://blog.csdn.net/m0_62780716/article/details/138391887

版权

力扣数据结构训练专栏收录该内容

12 篇文章 0 订阅

订阅专栏

该代码提供了一个算法，使用哈希表来计算给定整数数组中和为特定值k的连续子数组数量。时间复杂度为O(n)，空间复杂度为O(n)，适用于解决此类问题。

摘要由CSDN通过智能技术生成

class Solution {
public:
    int subarraySum(vector<int>& nums, int k) {
        unordered_map<int, int> mp;
        mp[0] = 1;
        int count = 0, pre = 0;
        for (auto& x : nums)
        {
            pre += x;
            if (mp.find(pre - k) != mp.end())
            {
                count += mp[pre - k];
            }
            mp[pre] ++;
        }
        return count;
    }
};

这段代码是一个解决特定问题的算法实现，即寻找和为特定值 k 的连续子数组的数量。下面是具体的步骤讲解：

unordered_map<int, int> mp; 创建一个哈希表，用于存储前缀和及其出现的次数。

mp[0] = 1; 初始化哈希表，将前缀和为 0 的计数设置为 1，这是为了处理当子数组的起始点在数组的第一个元素之前的情况。

int count = 0, pre = 0; 初始化计数器 count，用于记录和为 k 的子数组的数量，以及前缀和 pre。

for (auto& x : nums) 遍历输入数组 nums 中的每个元素。

pre += x; 更新当前的前缀和，将当前元素 x 的值加到 pre 上。

if (mp.find(pre - k) != mp.end()) 检查是否存在一个前缀和，使得 pre - (pre - k) = k，即从某个点到当前元素的和为 k。

count += mp[pre - k]; 如果存在这样的前缀和，说明找到了一个和为 k 的子数组，将这个前缀和对应的次数加到 count 上。

mp[pre] ++; 更新哈希表，将当前的前缀和 pre 的计数增加 1。

return count; 遍历完成后，返回计数器 count 的值，即和为 k 的子数组的数量。

这个算法的时间复杂度是 O(n)，其中 n 是数组 nums 的长度，因为算法只需要遍历数组一次。空间复杂度是 O(n)，因为需要存储前缀和的哈希表，最坏情况下前缀和的数量与数组的长度相同。