JavaScript 按规则数组对对象集合分组的实现与遍历逻辑选择

没有绝对最优的遍历选择，完全匹配当前业务的约束条件才是最优解，两种方式的适用场景非常明确：

• 遍历规则数组a的适用场景
  • 要求最终分组顺序严格和规则定义的顺序对齐，比如规则组1的所有内容必须排在结果数组的第一个位置，规则组2排在第二个位置
  • 不需要保留待分组集合中无规则归属的对象，自动过滤掉游离项
  • 规则组数量远少于待分组对象的数量（比如几十组规则、几十万条待分组数据），这种方式的整体计算开销最低
• 遍历待分组数组b的适用场景
  • 要求必须保留b中的所有对象，哪怕没有匹配到任何规则，也要将其归入默认分组或者单独成组
  • 待分组对象数量远少于规则数量（比如几十条待分组数据、几十万条规则），可以减少无效的规则遍历
  • 分组结果的顺序要求和待分组对象在b中的出现顺序一致，不需要严格对齐规则的排列顺序

针对示例中“严格匹配规则顺序、自动过滤b中不存在的规则元素”的需求，最简洁高效的实现是提前做id映射，避免嵌套循环带来的性能损耗：

// 1. 构建待分组对象的_id索引，单次遍历b完成构建，后续查询都是O(1)复杂度
const bIdToItemMap = b.reduce((map, currentItem) => {
  map[currentItem._id] = currentItem
  return map
}, {})

// 2. 按规则数组遍历生成结果，完全对齐a的顺序
const groupResult = a.map(ruleGroup => {
  // 取当前规则下的元素，过滤掉b中不存在的无效id
  return ruleGroup.elements
    .map(elementId => bIdToItemMap[elementId])
    .filter(Boolean)
})

这段代码的输出和题目给出的预期结果完全一致：第一组会过滤掉不存在的'sd'，剩下aasa、zzzx、cccx三个对象；第二组保留bb、xx两个对象。

如果需要适配“保留所有b中对象、无规则归属的项不丢弃”的场景，就选择遍历b的实现方式：

// 1. 建立规则id和所属分组索引的映射表
const ruleIdToGroupIndex = {}
a.forEach((groupRule, groupIndex) => {
  groupRule.elements.forEach(id => {
    ruleIdToGroupIndex[id] = groupIndex
  })
})

// 2. 初始化结果容器
const resultGroups = Array.from({ length: a.length }, () => [])
const ungroupedItems = [] // 存储没有匹配到规则的对象

// 3. 单次遍历b完成分组
b.forEach(item => {
  const targetGroupIndex = ruleIdToGroupIndex[item._id]
  if (targetGroupIndex !== undefined) {
    resultGroups[targetGroupIndex].push(item)
  } else {
    ungroupedItems.push(item)
  }
})

• 之前尝试用reduce直接遍历b收集全量元素的写法，本质上没有建立规则和待分组对象的关联，所以无法生成符合要求的分组结构。
• 尽量不要直接在嵌套循环里查找匹配项，比如遍历a的每个元素时再全量遍历b找对应id，当数据量较大时会出现明显的性能瓶颈；提前构建id映射的做法可以把整体时间复杂度从O(n*m)降到O(n+m)。
• 注意示例中规则包含的'sd'在b中不存在，这类无效项会被自动过滤，不需要额外报错打断流程。

内容的提问来源于stack exchange，提问作者Paul