LeetCode 热门100题-和为 K 的子数组-错题

嵌入式码喽2025-02-17 0:36

题目描述：

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

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

示例 1：

复制代码

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

cpp 复制代码

class Solution {
public:
    int subarraySum(vector<int>& nums, int k) {
        unordered_map<int, int> prefix_sum_map; // 前缀和出现次数
        prefix_sum_map[0] = 1; // 初始化为0，表示从数组开头到当前元素和为k的子数组存在
        int current_sum = 0; // 当前前缀和
        int count = 0; // 子数组的个数
        
        for (int num : nums) {
            current_sum += num; // 更新当前前缀和
            if (prefix_sum_map.find(current_sum - k) != prefix_sum_map.end()) {
                count += prefix_sum_map[current_sum - k]; // 找到符合条件的子数组
            }
            prefix_sum_map[current_sum]++; // 更新前缀和的出现次数
        }
        
        return count;
    }
};

和为k的子数组的概念：就是从某个数开始到某个数结束（也可以是某个数自己开始到它自己结束），他们的和为k。

那反过来，先把数组中所有元素先进行从0开始的累加。如示例进行累加之后将是：0，1，2，3

可以将原来的数组看作是一个数列{An}，求和后的数列看作是{Sn}，那么Sn-S(n-k)的结果就是从第(n-k)项开始到第n项的和，用Δ来接收这个数值，就是Δ=Sn-S(n-k)，如果Δ=target，则表示找到了一次和为target的子序列，子序列从第(n-k)开始到第n结束，由于此时n和k都是任意数，所以用这个公式可以适用于题目要求。

错题记录：

复制代码

class Solution {
public:
    int subarraySum(vector<int>& nums, int k) {
        int result=0;
        int sum=0;
        unordered_map<int,int> Sn;
        Sn[0]=1;
        for(const int& num:nums)
        {
            sum+=num;
            if(Sn.find(sum-k)!=Sn.end())   //注意是当前和-k  ，一开始给的k-sum了
            {
                result+=Sn[sum-k];
            }
            Sn[sum]++;
        }
        return result;
    }
};