JavaScript中Iterator可通过生成器递归或栈模拟实现嵌套结构惰性展平：生成器用yield*委托递归，栈则逆序压入元素防爆栈，均支持数组、树及混合结构，需注意循环引用、稀疏数组和null/undefined处理。
JavaScript 中的 Iterator 可以通过递归或栈模拟递归的方式，实现对多层嵌套数据结构（如嵌套数组、树、对象等）的展平迭代。核心思路是：**不一次性展开全部数据，而是按需逐个返回最内层的值**，保持惰性求值和内存友好。
用生成器函数递归实现嵌套数组展平
这是最直观、可读性最好的方式。生成器（function*）天然支持 yield* 语法，可委托子迭代器，自动处理递归展开。
示例：展平任意深度的嵌套数组（忽略非数组/非原始值）：
function* flatten(iterable) {
for (const item of iterable) {
if (Array.isArray(item)) {
yield* flatten(item); // 委托给子生成器，递归展开
} else {
yield item; // 基础值直接产出
}
}
}
<p>// 使用
const nested = [1, [2, [3, 4], 5], 6];
for (const val of flatten(nested)) {
console.log(val); // 1, 2, 3, 4, 5, 6
}
手动实现 Iterator 接口（无生成器时）
若需兼容旧环境或自定义控制逻辑（如跳过 null、支持对象属性遍历），可手动实现 [Symbol.iterator]() 方法，用栈模拟递归调用栈：
立即学习“Java免费学习笔记（深入）”；
- 初始化时将根数据推入栈
-
next()中弹出当前项；若是数组/类数组，将其元素逆序压入栈（保证顺序）；否则返回该值 - 用栈替代函数调用栈，避免爆栈风险
class FlattenIterator {
constructor(data) {
this.stack = Array.isArray(data) ? [...data] : [data];
}
<p>next() {
while (this.stack.length > 0) {
const item = this.stack.pop();
if (Array.isArray(item)) {
// 逆序压入，保持原顺序（如 [a,[b,c]] → 先压 c 再压 b）
for (let i = item.length - 1; i >= 0; i--) {
this.stack.push(item[i]);
}
} else {
return { value: item, done: false };
}
}
return { done: true };
}</p><p>[Symbol.iterator]() {
return this;
}
}</p><p>// 使用
const iter = new FlattenIterator([1, [2, [3, 4]], 5]);
for (const v of iter) console.log(v); // 1, 2, 3, 4, 5
扩展支持对象与混合结构
实际场景中嵌套结构常含对象（如树节点 { value, children }）。可在判断逻辑中加入对象分支：
- 检测
item && typeof item === 'object' && !Array.isArray(item) && 'children' in item - 若匹配，把
item.children（需确保是数组）压入栈或递归委托 - 也可统一用
Symbol.iterator检测：若item[Symbol.iterator]存在且为函数，则视为可迭代，用yield*或栈展开
示例（生成器版，支持数组 + 树对象）：
function* flattenMixed(item) {
if (Array.isArray(item)) {
for (const child of item) yield* flattenMixed(child);
} else if (item && typeof item === 'object' && 'children' in item) {
yield item.value; // 先产出当前节点值
if (Array.isArray(item.children)) {
for (const child of item.children) yield* flattenMixed(child);
}
} else {
yield item;
}
}
注意边界与性能细节
展平迭代器易忽略的要点：
-
循环引用：嵌套结构若存在环（如
a.b = a），必须加访问标记（如 WeakSet 记录已遍历对象），否则无限循环 -
稀疏数组与类数组：用
for...of自动跳过空槽；手动实现时建议用Array.isArray()+item.length判断，而非instanceof Array -
原始值与 null/undefined：明确是否产出它们（如
null是有效值还是应跳过），通常基础类型直接yield，null/undefined可选择跳过 - 只读保障：迭代器本身不修改原数据，但若传入的是可变引用（如嵌套对象），外部修改会影响后续迭代结果——这是预期行为，无需干预
