javascript - 展平嵌套数组后保留元素的索引

Question

我有一个单级嵌套数组，如下所示。

var nestedArrays = [[0, 1, 2, 3], [4, 5, 6], [7, 8, 9]];

我使用函数将其展平为单个统一数组( by Michal Perlapowski in another question here ，并且比通常的reduce/concat更快，由于兼容性问题无法使用 native flat() ):

const flatten = function(arr, result = []) {
  for (let i = 0, length = arr.length; i < length; i++) {
    const value = arr[i];
    if (Array.isArray(value)) {
      flatten(value, result);
    } else {
      result.push(value);
    }
  }
  return result;
};
...
merged = flatten(nestedArrays);

这样我就可以得到下面的内容:

[0,1,2,3,4,5,6,7,8,9]

现在说，我有值“3”，我希望找到它来自nestedArrays 中的哪个数组。 (例如，在“3”的情况下，返回值“0”，因为它来自nestedArrays 中的第0 个元素数组)。我该如何最简单或最有效地解决这个问题？

我尝试了几种方法来实现这项工作，例如:

1) 将现有嵌套数组转换为每个嵌套值的一对值，其中包含值本身和嵌套数组在父数组中的索引。简单但重复的数据，创建中间数组和额外的 forEach 循环:

var nestedArrays2 = [[0,0],[1,0],[2,0],[3,0],[4,1],[5,1],[6,1],[7,2],[8,2],[9,2]];

然后，同样将其压平，但这将导致:

[0,0,1,0,2,0,3,0,4,1,5,1,6,1,7,2,8,2,9,2]

也可以过滤奇数值，总而言之:

merged = flatten(nestedArrays2).filter((num, index) => {
  return index % 2 !== 0;
})

因此，在这种情况下，我知道 nestedArrays2[merged[i]][1] 将返回原始嵌套数组的索引，即 merged[i] 处的值就像在 nestedArrays 中一样。

1a) 从前一个分支中分支出来，但总体思路相同，可以创建一个单独的数组，而无需更改原始 nestedArrays。由于我们已经循环遍历它，因此我们不妨花时间在同一次迭代中将其展平。

var merged = [];
var nestIndexes = [];
nestedArrays.forEach((element,index){
  element.forEach((e) {
    merged.push(e);
    nestIndexes.push(index);
  });
});

这将导致合并索引和嵌套索引，如下所示:

merged [0,1,2,3,4,5,6,7,8,9]
nestedIndexes [0,0,0,0,1,1,1,3,3,3]

因此，如果我在 merged 中获取某个值的 index，我可以立即使用 从 nestedIndexes 获取等效值嵌套索引[索引]

2)保留单独的索引引用数组

我想也许我可以在合并嵌套数组之前保留一个单独的数组来标记嵌套数组开始的索引位置。只需循环 nestedArrays，获取每个元素的 length 并添加到前一个值，以便它是累积的:

var nestPositions = [];
nestedArrays.forEach((element, index) {
  nestPositions[index] = (element.length + (nestedArrays[index - 1] || 0) -1); //for when index is 0
})

因此，nestPositions 将是:

[3,6,9]

nestedArrays 中每个嵌套数组的最后一个元素的索引。

但在这种情况下，在我得到 merged[i] 后，我必须每次都循环遍历 NestPositions 数组，以查看是否小于或等于 i 并中断，以便检查每个元素.

3) 使用 nestedArrays 中的嵌套数组索引创建映射。

nestMap = new Map();
nestedArrays.forEach((element, index) => {
  nestMap.set(index, element);
})

然后我可以正常合并，并从 merged[i] 获取元素的嵌套数组索引，我可以假设，在 map 上执行 forEach，如下所示:

nestMap.forEach((value, key) => {
  if (value.includes(merged[i]) return key;
}

总之，这就是我脑子里能想到的方法。哪一个是最好的，是否有更好更简洁的方法来解决这个问题？不管怎样，感谢任何帮助，甚至只是为了达到这一部分哈哈。

Answer 1

在 Map 上执行 forEach 违背了使用 Map 的目的。但是，如果您创建一个从元素值到其数组索引的 Map，您可以直接使用 map.get(value) 获得答案:

var nestedArrays = [[0, 1, 2, 3], [4, 5, 6], [7, 8, 9]];
var merged = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9];

var nestedIndex = new Map();
nestedArrays.forEach((nested, i) => {
  nested.forEach((element, j) => nestedIndex.set(element, i));
});

console.log(nestedIndex.get(merged[0]));
console.log(nestedIndex.get(merged[3]));
console.log(nestedIndex.get(merged[5]));
console.log(nestedIndex.get(merged[9]));