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

题目描述:

给你一个整数数组 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;
    }
};
相关推荐
Dizzy.51720 分钟前
数据结构(查找)
数据结构·学习·算法
分别努力读书3 小时前
acm培训 part 7
算法·图论
武乐乐~3 小时前
欢乐力扣:赎金信
算法·leetcode·职场和发展
Jared_devin3 小时前
数据结构——模拟栈例题B3619
数据结构
'Debug3 小时前
算法从0到100之【专题一】- 双指针第一练(数组划分、数组分块)
算法
sushang~4 小时前
leetcode21.合并两个有序链表
数据结构·链表
Fansv5874 小时前
深度学习-2.机械学习基础
人工智能·经验分享·python·深度学习·算法·机器学习
yatingliu20195 小时前
代码随想录算法训练营第六天| 242.有效的字母异位词 、349. 两个数组的交集、202. 快乐数 、1. 两数之和
c++·算法
uhakadotcom5 小时前
Google DeepMind最近发布了SigLIP 2
人工智能·算法·架构
sjsjs115 小时前
【数据结构-并查集】力扣1202. 交换字符串中的元素
数据结构·leetcode·并查集