LeetCode热题100--560.和为K的子数组(前缀和)--中等

1.题目

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

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

示例 1:

输入:nums = [1,1,1], k = 2

输出:2

示例 2:

输入:nums = [1,2,3], k = 3

输出:2

2. 题解

java 复制代码
class Solution {
    public int subarraySum(int[] nums, int k) {
        int n = nums.length;
        int[] s = new int[n+1];
        for(int i = 0; i < n; i++){
            s[i+1] = s[i] + nums[i];
        }

        int ans = 0;
        Map<Integer,Integer> cnt = new HashMap<>(n+1);
        for (int sj : s){
            ans += cnt.getOrDefault(sj - k, 0);
            cnt.merge(sj, 1, Integer::sum);
        }
        return ans;
    }
}

3. 解析

出自这位老师:灵茶山艾府:前缀和+哈希表:从两次遍历到一次遍历,附变形题(Python/Java/C++/Go/JS/Rust)

第1部分:初始化变量和前缀和数组

java 复制代码
int n = nums.length;
int[] s = new int[n+1];
  • int n = nums.length;:获取输入数组nums的长度,赋值给n。
  • int[] s = new int[n+1];:创建一个长度为n+1的前缀和数组s。前缀和数组用于存储从数组开头到当前位置的累加和。

第2部分:填充前缀和数组

java 复制代码
for(int i = 0; i < n; i++){
    s[i+1] = s[i] + nums[i];
}
  • 使用一个循环从i=0遍历到nums.length-1,计算前缀和。
  • s[0]初始化为0,表示前缀和数组的第一个元素是0。
  • 对于每个索引i(0 ≤ i < n),将当前nums的值加到s[i+1]中。这样,s[i+1]就记录了从nums[0]到nums[i]的累加和。

第3部分:初始化答案变量

java 复制代码
int ans = 0;
  • int ans = 0;:用于累计符合条件子数组的数量。

第4部分:创建哈希表并初始化

java 复制代码
Map<Integer,Integer> cnt = new HashMap<>(n+1);
  • 创建一个容量为n+1的HashMap,键值类型是Integer。
  • hashmap的作用是统计每个前缀和出现的次数。这里预先分配了n+1个空间。

第5部分:遍历前缀和数组

java 复制代码
for (int sj : s){
    ans += cnt.getOrDefault(sj - k, 0);
    cnt.merge(sj, 1, Integer::sum);
}
  • 遍历s数组中的每一个元素sj。
  • 对于每个sj,首先查找键值对(sj - k)在map中出现的次数,并将这个次数加到ans中。这一步的作用是找到所有以当前前缀和sj为结尾且子数组和等于k的子数组数量。
  • 然后更新哈希表cnt:将键sj的出现次数增加1。

第6部分:返回答案

java 复制代码
return ans;

最终返回累计的符合条件子数组的数量ans。

相关推荐
断剑zou天涯13 小时前
【算法笔记】暴力递归尝试
java·笔记·算法
油泼辣子多加14 小时前
【实战】自然语言处理--长文本分类(1)DPCNN算法
算法·自然语言处理·分类
I'm a winner15 小时前
基于YOLO算法的医疗应用专题:第一章 计算机视觉与深度学习概述
算法·yolo·计算机视觉
vir0215 小时前
P1928 外星密码(dfs)
java·数据结构·算法·深度优先·1024程序员节
喜欢吃燃面15 小时前
数据结构算法题:list
开发语言·c++·学习·算法·1024程序员节
寂静山林15 小时前
UVa 12991 Game Rooms
算法·1024程序员节
Java_小白呀16 小时前
第十四届蓝桥杯大赛软件赛国赛Java大学C组(部分)
职场和发展·蓝桥杯·1024程序员节
Dream it possible!16 小时前
LeetCode 面试经典 150_链表_合并两个有序链表(58_21_C++_简单)
leetcode·链表·面试·1024程序员节
余俊晖16 小时前
RLVR训练多模态文档解析模型-olmOCR 2技术方案(模型、数据和代码均开源)
人工智能·算法·ocr·grpo
凉虾皮17 小时前
2024包河初中组
学习·算法·1024程序员节