17370845950

JavaScript迭代器是统一遍历机制，要求对象实现Symbol.iterator方法返回含next()的迭代器对象，next()返回{value, done}；手写Range类和BinaryTree树结构示例展示其应用，生成器函数是最简洁实现方式。

JavaScript 迭代器是一种统一的遍历机制，它让任意数据结构只要满足特定协议（即实现 [Symbol.iterator]() 方法），就能被 for...of、扩展运算符（...）、解构赋值等语法自然消费。

迭代器的核心：Iterator 接口

一个对象要成为可迭代的，必须在其原型链上提供一个名为 Symbol.iterator 的方法，该方法返回一个**迭代器对象**。这个迭代器对象必须有 next() 方法，每次调用返回形如 { value: any, done: boolean } 的对象。

• value：当前迭代出的值（可以是任意类型）
• done：布尔值，true 表示遍历结束，false 表示还有值可取

手写一个简单迭代器

比如想让一个自定义的区间类 Range 支持 for...of：

class Range {
  constructor(start, end) {
    this.start = start;
    this.end = end;
  }
[Symbol.iterator]() {
let current = this.start;
return {
next: () => {
if (current <= this.end) {
return { value: current++, done: false };
} else {
return { value: undefined, done: true };
}
}
};
}
}
// 使用
for (const n of new Range(2, 5)) {
console.log(n); // 2, 3, 4, 5
}

为更复杂的数据结构添加迭代能力

例如树结构，常需支持多种遍历顺序（前序、中序、后序、层序）。可将不同策略封装为独立的迭代器方法：

class TreeNode {
  constructor(val, left = null, right = null) {
    this.val = val;
    this.left = left;
    this.right = right;
  }
}
class BinaryTree {
constructor(root) {
this.root = root;
}
[Symbol.iterator]() {
// 默认提供中序遍历（生成器写法更简洁）
function inorder(node) {
if (!node) return;
yield inorder(node.left);
yield node.val;
yield inorder(node.right);
}
yield* inorder(this.root);
}
// 额外提供前序遍历方法
preorder() {
function traverse(node) {
if (!node) return;
yield node.val;
yield traverse(node.left);
yield* traverse(node.right);
}
return traverse(this.root);
}
}
// 使用
const tree = new BinaryTree(
new TreeNode(1,
new TreeNode(2),
new TreeNode(3)
)
);
for (const val of tree) {
console.log(val); // 2, 1, 3 （中序）
}
for (const val of tree.preorder()) {
console.log(val); // 1, 2, 3 （前序）
}

注意：function* 生成器函数会自动返回符合迭代器协议的对象，是实现迭代器最常用、最简洁的方式。

常见误区与注意事项

• 迭代器是一次性的：多次调用 [Symbol.iterator]() 应返回**新**的迭代器对象，否则 for...of 多次执行会“失效”
• done: true 后再调用 next()，仍应返回 { value: undefined, done: true }
• 数组、Map、Set 等内置类型已原生支持迭代器；类数组对象（如 arguments）需手动部署才能被 for...of 正确识别
• 不要在迭代过程中修改被遍历结构本身（如边遍历边删除节点），行为未定义且易出错