说在前面
🎈不知道大家对于算法的学习是一个怎样的心态呢?为了面试还是因为兴趣?不管是出于什么原因,算法学习需要持续保持。
题目描述
请你编写一个函数,它接收一个 多维数组 arr 和它的深度 n ,并返回该数组的 扁平化 后的结果。
多维数组 是一种包含整数或其他 多维数组 的递归数据结构。
数组 扁平化 是对数组的一种操作,定义是将原数组部分或全部子数组删除,并替换为该子数组中的实际元素。只有当嵌套的数组深度大于 n 时,才应该执行扁平化操作。第一层数组中元素的深度被认为是 0。
请在没有使用内置方法 Array.flat 的前提下解决这个问题。
示例 1:
输入 arr = [1, 2, 3, [4, 5, 6], [7, 8, [9, 10, 11], 12], [13, 14, 15]] n = 0 输出 [1, 2, 3, [4, 5, 6], [7, 8, [9, 10, 11], 12], [13, 14, 15]] 解释 传递深度 n=0 的多维数组将始终得到原始数组。这是因为 子数组(0) 的最小可能的深度不小于 n=0 。因此,任何子数组都不应该被平面化。
示例 2:
输入 arr = [1, 2, 3, [4, 5, 6], [7, 8, [9, 10, 11], 12], [13, 14, 15]] n = 1 输出 [1, 2, 3, 4, 5, 6, 7, 8, [9, 10, 11], 12, 13, 14, 15] 解释 以 4 、7 和 13 开头的子数组都被扁平化了,这是因为它们的深度为 0 , 而 0 小于 1 。然而 [9,10,11] 其深度为 1 ,所以未被扁平化。
示例 3:
输入 arr = [[1, 2, 3], [4, 5, 6], [7, 8, [9, 10, 11], 12], [13, 14, 15]] n = 2 输出 [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15] 解释 所有子数组的最大深度都为 1 。因此,它们都被扁平化了。
提示:
- 0 <= arr 的元素个数 <= 105
- 0 <= arr 的子数组个数 <= 105
- maxDepth <= 1000
- -1000 <= each number <= 1000
- 0 <= n <= 1000
解题思路
在函数中,首先创建一个空数组 res 用于存放扁平化后的结果。
然后定义内部函数 flatFn,它使用 for 循环遍历数组中的每个元素:
- 如果当前元素是数组,并且递归深度小于 n,则递归调用 flatFn 处理该子数组,并将深度加 1。
- 否则,将元素直接压入 res 数组中。
最后,调用 flatFn 函数,并以初始深度 0 作为第二个参数,将扁平化后的结果返回。
AC代码
/** * @param {Array} arr * @param {number} depth * @return {Array} */ var flat = function (arr, n) { const res = []; const flatFn = (arr, depth) => { for (let i = 0; i < arr.length; i++) { if (Array.isArray(arr[i]) && depth < n) { flatFn(arr[i], depth + 1); } else { res.push(arr[i]); } } }; flatFn(arr, 0); return res; };
公众号
关注公众号『前端也能这么有趣
』,获取更多有趣内容。
说在后面
🎉 这里是 JYeontu,现在是一名前端工程师,有空会刷刷算法题,平时喜欢打羽毛球 🏸 ,平时也喜欢写些东西,既为自己记录 📋,也希望可以对大家有那么一丢丢的帮助,写的不好望多多谅解 🙇,写错的地方望指出,定会认真改进 😊,偶尔也会在自己的公众号『
前端也能这么有趣
』发一些比较有趣的文章,有兴趣的也可以关注下。在此谢谢大家的支持,我们下文再见 🙌。