本文主要是介绍【数组】-Lc325-和等于k的最长子数组长度(前缀和 + Map),希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
写在前面
最近想复习一下数据结构与算法相关的内容,找一些题来做一做。如有更好思路,欢迎指正。
目录
- 写在前面
- 一、场景描述
- 二、具体步骤
- 1.环境说明
- 2.概念
- 2.1 什么是子数组
- 2.2 前缀和
- 3.关键点
- 3.1 map初始化问题
- 3.2 map的key重复,取舍问题
- 4.代码
- 写在后面
一、场景描述
和等于k的最长子数组长度。
给定一个数组 nums 和一个目标值 k,找到和等于 k 的最长子数组长度。如果不存在任意一个符合要求的子数组,则返回 0。
示例 1:
输入: nums = [1, -1, 5, -2, 3], k = 3
输出: 4
解释: 子数组 [1, -1, 5, -2] 和等于 3,且长度最长。示例 2:
输入: nums = [-2, -1, 2, 1], k = 1
输出: 2
解释: 子数组 [-1, 2] 和等于 1,且长度最长
二、具体步骤
1.环境说明
名称 | 说明 |
---|---|
IntelliJ IDEA | 2019.2 |
2.概念
2.1 什么是子数组
子数组 ≠ 子集
子数组 ≠ 原数组中随机选几个元素组成的数组
子数组是原数组中连续的元素组成的数组。
2.2 前缀和
设 sum[i
] 表示 nums[0] + nums[1] + … + nums[i-1
] 的和,称为第 i
位的前缀和。
于是,如果存在两个索引 i 和 j,使得 sum[j] - sum[i] == k,
说明找到一个子数组 [i, j-1] ,子数组的和为 k。
那么,2个前缀和相减,中间的值等于k,那么中间的这部分就是子数组
。
转化,
定义一个 map,key为 sum[i],value为 i,
那么sum[j] - k,判断能否在 map 中找到?如果找得到,即存在这样的 k。
3.关键点
3.1 map初始化问题
Map<Integer, Integer> map = new HashMap<Integer, Integer>(){{put(0, -1);
}};
需要初始化 map,key = 0, value = -1,为 0 赋值默认值。
这个貌似是属于技巧性的,目前也不是太理解😅,还不能从原理上解释。
分析:数组 {1, -1, 5, -2, 3},k=3
说明: j 为索引,num[j]为数组中元素,sum为前缀和,preSum为sum-k的值。j num[j] sum preSum
——————————————————————
-1 00 1 1 -21 -12 5 5 23 -2 3 04 3 6 3结果:3 - (-1) = 4
3.2 map的key重复,取舍问题
map.put(sum, j),
注意需要加一个判断条件,key可能会重复,也就是sum[i]和sum[某一个值]相同。
因为取 len 最长,所以保留原始i值,即添加条件是 !map.containsKey(sum)
示例说明:原数组:[5, -2, -3, 3]
前缀和:[5, 3, 0, 3],同时存在两个3,因为要求最长序列,所以map中保存的 index=1的sum=3,而不是index=3的sum=3
index: 0 1 2 3
4.代码
以下为Java版本实现:
public class Lc325_maxSubArrayLen {public static void main(String[] args) {int[] nums = {1, -1, 5, -2, 3};System.out.println(maxSubArrayLen(nums, 3)); // 4
// int[] nums = {-2, -1, 2, 1};
// System.out.println(maxSubArrayLen(nums, 1)); // 2}/*** 什么是子数组?* 子数组是原数组中连续的元素组成的数组,不是随机选几个元素组成的数组** 返回值是 int* 思路: 前缀和 + Map** 定义:* 设 sum[i] 表示 nums[0] + nums[1] + … + nums[i-1] 的和,称为第 i 位的前缀和。* 于是,如果存在两个索引 i 和 j,使得 sum[j] - sum[i] == k,说明找到一个子数组 [i, j-1] ,子数组的和为 k* 那么,2个前缀和相减,中间的值等于k,那么中间的这部分就是子数组** 转化:* 定义一个 map,key为 sum[i],value为 i,* 那么sum[j] - k,判断能否在 map 中找到,如果找得到,即存在这样的 k** 定义一个 map 存储 sum[i],* 定义一个 sum=0累加,len=0为最大长度** for循环nums,int j = 0; j < nums.length; j++* sum += nums[i]* int preSum = sum - k** 判断 map.containsKey(preSum)* 如果包含那么len就取 Math.max(len, j - map.get(preSum))** map.put(sum, j), 注意需要加一个判断条件:key可能会重复,也就是sum[i]和sum[某一个值]相同* 因为取len最长,所以保留原始i值,即添加条件是 !map.containsKey(sum)** 示例说明:* 原数组:[5, -2, -3, 3]* 前缀和:[5, 3, 0, 3], 同时存在两个3,因为要求最长序列,所以map中保存的 index=1的sum=3,而不是index=3的sum=3* index: 0 1 2 3*/private static int maxSubArrayLen(int[] nums, int k) {/*** 注意:* 需要初始化 map,key = 0, value = -1,为 0 赋值默认值*** 数组 {1, -1, 5, -2, 3}** j num[j] sum preSum* -1 0* 0 1 1 -2* 1 -1 0 -3* 2 5 5 2* 3 -2 3 0* 4 3 6 3*/Map<Integer, Integer> map = new HashMap<Integer, Integer>(){{put(0, -1);}};int len = 0, sum = 0;for (int j = 0; j < nums.length; j++) {sum += nums[j];int preSum = sum - k;if (map.containsKey(preSum)) {len = Math.max(len, j - map.get(preSum));}if (!map.containsKey(sum)) {map.put(sum, j);}}return len;}
}
写在后面
如果本文内容对您有价值或者有启发的话,欢迎点赞、关注、评论和转发。您的反馈和陪伴将促进我们共同进步和成长。
这篇关于【数组】-Lc325-和等于k的最长子数组长度(前缀和 + Map)的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!