LeetCode hoot 100 -- 和为K的子数组

本题取自LeetCode hoot 100 题号560 和为K的子数组

一 题目概述

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

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

二 思路解析

题中提到子数组数连续序列，我们容易想到利用'前缀和'。'前缀和'顾名思义，前n个元素的和。观察其特性不难发现，原数组中任意一个连续序列的和都可以由某两个前缀和相减得到。示意图如下：

经过前缀和的转换，此题是否就可以简化的看成"两数之和"的问题了？也就是说遍历所有前缀和curr_sum，判断该curr_sum之前的前缀和中是否出现了值curr_sum。值得一提，同一个值可能对应多个前缀和，如：[1,1,0]，又如：[1,-1,1] 。这样也会使得和为K的数组结果变多。所以在此我们不妨利用unordered_map结构，既能够做到快速查找，又能够记录某一个值出现的次数。

三 代码解析

// LeetCode 模型
class Solution {
public:
    // 传入数组，和K值
    int subarraySum(vector<int>& nums, int k) {
        // 利用unordered_map结构，记录前缀和及其出现的次数
        unordered_map<int, int>hash;
        // 注意hash[0]=1(关键）处理当某前缀和就等于K时的情况
        hash[0] = 1;
        int curr_sum = 0;// 当前的前缀和
        int res = 0;// 结果次数
    
        for (int num : nums)
        {
            // curr_sum记录目前的前缀和
            curr_sum += num;
            int target = curr_sum - k;
            if (hash.find(target) != hash.end())
            {
                res += hash[target];
            }
            // hash记录在此前缀和之前 的前缀和出现的次数
            hash[curr_sum]++;
        }
        return res;
    }
};

以上就是本文所有内容了。可以看到封面刷题数字由'12'变为'13'了，大家可以一起打卡保持'每日一题'的习惯。最后感谢观看。