一道关于 ARRAY 深度展开的面试题

昨天面试的时候,考官出了这么一个题,写一个函数,输入a = [1,[2,3,[4,5,6]]],输出a = [1,2,3,4,5,6]。当时我脑子有点短路,做了好久,给出了个不太符合要求的答案如下

var a = [1,[2,3,[4,5,6]]]
var arr = []
function merge(a) {
  for(var i = 0; i < a.length; i++){
    if(Array.isArray(a[i])){
      merge(a[i])
    } else {
      arr.push(a[i])
    }
  }
  return arr
}

回家后稍微完善了下,符合了考官输入输出的要求了(不过也晚了)

function flatten(a) {
  if (!Array.isArray(a)) return

  var arr = []

  return (function merge(a) {
    for(var i = 0, l = a.length; i < l; i++){
      if(Array.isArray(a[i])){
        merge(a[i])
      } else {
        arr.push(a[i])
      }
    }
    return arr
  })(a)
}

然后跟朋友聊天问问有啥简单的方法,朋友给了一个挺符合这个题的简单方法类似如下

const flatten = (arr) => arr.toString().split(',').map(v => Number(v))

不过这个答案只适用于这道题,如果不是全数字的就不行了。于是在GitHub上搜索array flatten,看了看排名前三的源码,发现有好多不同的写法,比如

function flatten() {
  var args = Array.from(arguments)
  do {
    args = [].concat.apply([], args)
  } while (args.some(Array.isArray))
  return args
}

也有这么写的

function flatten (array) {
  var result = Array.prototype.concat.apply([], array)
  if (result.length === array.length) {
    return result
  }
  return flatten(result)
}

写法各不相同,还有好多看似类似,但细节上不一样的,所以关键就看性能了,好在arr-flattenarray-flatten中就都有benchmark,进行了各种比较

arr-flatten这个项目中,在各种不同的、可以在 npm 上搜到的array flatten实现方法之间进行了比较,最后的结论是array-flatten执行的效率最高

array-flatten项目里的benchmark主要是自己源码中细节不同的写法之间的差异,最后得出的最佳实践如下

function flatten (array) {
  if (!Array.isArray(array)) {
    throw new TypeError('Expected value to be an array')
  }
  return flattenFrom(array)
}

function flattenFrom (array) {
  return flattenDown(array, [])
}

function flattenDown (array, result) {
  for (var i = 0; i < array.length; i++) {
    var value = array[i]

    if (Array.isArray(value)) {
      flattenDown(value, result)
    } else {
      result.push(value)
    }
  }

  return result
}

这套代码思路与我给出的答案是一样的,但由于细节上的不同,导致用benchmark测试,执行效率比我写的那个快了约20%

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 背景 一年多以前我在知乎上答了有关LeetCode的问题, 分享了一些自己做题目的经验。 张土汪:刷leetcod...
    土汪阅读 12,768评论 0 33
  • <a name='html'>HTML</a> Doctype作用?标准模式与兼容模式各有什么区别? (1)、<...
    clark124阅读 3,544评论 1 19
  • 数组是一种可变的、可索引的数据集合。在Scala中用Array[T]的形式来表示Java中的数组形式 T[]。 v...
    时待吾阅读 967评论 0 0
  • 今天是儿童节,可我都三十了,已经不算儿童了,对于这样的节日确实没什么好说的。昨天听老梁的专题又讲到了日本的痴文化~...
    韧针小郭阅读 333评论 0 0
  • 10.2 一个巧合,一段故事 我们曾认为只是彼此的过客 无数次的擦肩而过,我却没有曾为你停下我的跳步 我们曾生活在...
    YC一辰阅读 257评论 1 0