本文介绍一种递归算法,用于将具有树状层级结构的嵌套数组(每层为若干子数组)展开为所有合法路径字符串,核心是按层级顺序“逐段拼接”,并精确控制各层子数组的选取偏移。
本文介绍一种递归算法,用于将具有树状层级结构的嵌套数组(每层为若干子数组)展开为所有合法路径字符串,核心是按层级顺序“逐段拼接”,并精确控制各层子数组的选取偏移。
该问题本质是多级笛卡尔展开的变体:不同于标准笛卡尔积(每一层所有元素与下一层所有元素两两组合),此处要求上一层的每个结果项,仅与下一层中对应序号的子数组进行组合——即第 i 个中间结果,必须与第 i 个子数组(而非全部子数组)进行拼接。这种结构天然对应一棵多叉树的根到叶路径枚举,其中每层的子数组数量决定了该层各节点的分支数。
关键难点在于:不能简单对每层做全量笛卡尔积,而需维护一个“层级偏移指针序列”,确保路径生成时严格遵循“父节点索引 → 子数组索引 → 子元素”的映射关系。
以下是一个健壮、可读性强的实现方案:
function expandTreeBranches(data) {
// 递归主函数:index 表示当前处理层级,offsets 记录各层已使用的子数组索引
function combine(index = 0, offsets = []) {
// 基础情况:已遍历完所有层级,返回空字符串作为拼接起点
if (index >= data.length) return [''];
// 获取当前层级的“源数组”:
// - 若 index === 0,直接取 data[0](顶层字符串数组)
// - 否则,取 data[index][offsets[index]](对应父路径索引所选的子数组)
const currentSource = index === 0
? data[0]
: data[index][offsets[index]];
// 对 currentSource 中每个元素,递归生成后续路径,并拼接
return currentSource.flatMap((item, i) => {
// 为下一层准备偏移数组:复制当前 offsets,并设置下一层初始偏移为 0
const nextOffsets = [...offsets];
nextOffsets[index] = i; // 记录当前层选用的是第 i 个子数组(仅对 index > 0 有意义)
// 递归获取后续路径,再与当前 item 拼接
return combine(index + 1, nextOffsets).map(suffix => item + suffix);
});
}
return combine();
}
但上述实现存在冗余拷贝。更优解是采用闭包状态管理偏移(如原答案所示),兼顾性能与简洁性:
const expandTreeBranches = (data) => {
const combine = (index = 0, offsets = [0]) => {
// 确保 offsets 至少有 index+1 个元素,初始化下一层偏移为 0
if (offsets.length = data.length) return [''];
// 当前层级的数据源
const source = index === 0
? data[0]
: data[index][offsets[index]];
// 对 source 中每个元素,递归生成后缀并拼接
return source.flatMap((item, idx) => {
// 更新下一层偏移:指向当前子数组的下一个兄弟(用于后续递归)
offsets[index + 1]++;
// 递归获取后缀,拼接返回
return combine(index + 1, offsets).map(suf => item + suf);
});
};
return combine();
};
// 测试用例
const input = [
["a", "b"],
[["c", "d"], ["e", "f", "g"]],
[["h", "i"], ["j"], ["k", "l"], ["m"], ["n", "o", "p"]]
];
console.log(expandTreeBranches(input));
// 输出: ["ach","aci","adj","bek","bel","bfm","bgn","bgo","bgp"]
✅ 注意事项:
- 该算法假设输入结构合法:data[0] 必须是字符串数组;data[i](i ≥ 1)必须是数组的数组,且其长度 ≥ data[i-1].length(否则某条路径会因子数组缺失而中断)。
- offsets 数组隐式记录了当前正在构建的路径在每一层所选择的子数组索引,offsets[index] 表示第 index 层选用的是 data[index] 中的第几个子数组。
- 使用 flatMap 替代 map + flat,语义更清晰且避免嵌套数组。
- 时间复杂度为 O(N),其中 N 为最终输出字符串总字符数;空间复杂度取决于最大递归深度和临时数组开销。
此方法精准建模了树形分支的展开逻辑,适用于配置生成、测试用例组合、路径枚举等场景。











