嵌套列表的Javascript长度

2023年4月19日 218次阅读

目标：给定一组混合类型确定每个级别的元素数量.如果在同一级别有两个子阵列,则每个子元素都会计入该级别的元素总数.

做法：

Array.prototype.elementsAtLevels = function( level, levelData ) {
  if ( level == undefined ) { level = 0;  } else { level += 1 }
  if ( levelData == undefined ) { levelData = {}; }
  if ( levelData[level] == undefined ) { levelData[level] = this.length} else { levelData[level] += this.length }
  this.map(function(e, i) {if (Array.isArray(e)){ e.elementsAtLevels(level, levelData) }})
  return levelData
}

测试用例：

[
  1,      // 0: 1
  1,      // 0: 2
  1,      // 0: 3
  1,      // 0: 4
  [       // 0: 5
    2,    // 1: 1
    2,    // 1: 2
    2     // 1: 3
  ], 
  [       // 0: 6
    [     // 1: 4
      3,  // 2: 1
      3   // 2: 2
    ],   
    [     // 1: 5
      [   // 2: 3
        4 // 3: 1
      ]
    ]
  ]
].elementsAtLevels()

// Object [ 6, 5, 3, 1 ]

题：
有更有效的方法来计算吗？

最佳答案 我写了一些非常类似于你所拥有的东西,并且在一个非常基本的基准测试中,它的运行时间不到一半.

let a = [1,1,1,1,[2,2,2],[[3,3],[[4]]]];

Array.prototype.elementsAtLevels2 = function (level, lData) {
    if (!level || !lData) {
        level = 0;
        lData = {};
    }

    if (!(level in lData)) {
        lData[level] = this.length;
    } else {
        lData[level] += this.length;
    }
    this.forEach(function (v) {
        if (Array.isArray(v))
            v.elementsAtLevels2(level + 1, lData);
    });

    return lData;
}

console.log(a.elementsAtLevels2());

我猜测主要的性能提升可能来自forEach vs map,map会创建一个新数组,而forEach则没有.

编辑

这是在JSBin

点赞