实现多维数组扁平化

如[1,2,[3,4,[5]]] ---> [1,2,3,4,5]方法1,使用Arrary.flat()方法实现

[[1], [2, [3, [4, 5]]]].flat(Infinity) 
// [1,2,3,4,5]
[[1], [2, [3, [4, 5]]]].flat(1) 
// 传入降维的数字得到想要实现的数组
// [ 1, 2, [ 3, [ 4, 5 ] ] ]

方法2,自己通过递归实现数组扁平化。通过递归方式对数组遍历,每次遍历判断元素是否是Array,是继续遍历,否返回元素,遍历结束即获得一个新的数组,例子如下:

// 通过map实现
function flat(arr) {
  let res = [];
  res = arr.map(value => {
    if (Array.isArray(value)) {
      return flat([...value], deep - 1);
    } else {
      return value;
    }
  });

  return [].concat(...res);
}
//通过reduce实现
function flat2(arr, deep = 1) {
  let res = [];
  if (deep > 0) {
    res = arr.reduce((pre, cur) => {
      return Array.isArray(cur)
        ? pre.concat(flat2(cur, deep - 1))
        : pre.concat(cur);
    }, []);
  } else {
    res = arr.slice();
  }

  return res;
}

实现Array数组扁平化方法。通过对flat2方法改进,传入一个deep参数控制遍历深度,从而控制降维的深度,实现如下:

function myFlat(arr, deep = 1) {
  if (arr === null || !arr.length) return [];
  let res = [];
  if (deep > 0) {
    res = arr.reduce((acc, cur) => {
      return Array.isArray(cur)
        ? acc.concat(myFlat(cur, deep - 1))
        : acc.concat(cur);
    }, []);
  } else {
    res = arr.slice();
  }

  return res;
}

const arr = [1, 2, 3, [1, 2, 3, 4, [2, 3, 4]]];

console.log(myFlat(arr, Infinity)); // [1,2,3,4,5]

console.log(myFlat(arr, 1)); // [ 1, 2, [ 3, [ 4, 5 ] ] ]

以上就是自己面试过程中遇到的问题,记录下来分享给大家。

你可能感兴趣的:(算法,javascript)