定义节点:
// 树节点定义
class TreeNode{
constructor(value){
this.value = value;
this.left = null;
this.right = null;
}
}
构建树:
// 构建树
let root = new TreeNode(1);
root.left = new TreeNode(2);
root.right = new TreeNode(3);
root.left.left = new TreeNode(4);
root.left.right = new TreeNode(5);
递归算法:
// 递归中序遍历二叉树
function midOrder(root) {
if(!root || !(root instanceof TreeNode)){
return;
}
// 递归访问左子树
midOrder(root.left);
console.log(root.value);
// 递归访问右子树
midOrder(root.right);
}
midOrder(root);
结果:
非递归算法:
// 非递归中序遍历二叉树
function midOrderN(root) {
let p = root; // p为当前遍历的节点, 初始为根
let arr = []; // arr作为栈
while(p || arr.length !== 0){
if(p){
// 遍历左子树
arr.push(p);
// 每遇到非空二叉树先向做走
p = p.left;
}else{
// p为空,出栈
let node = arr.pop();
// 访问该节点
console.log(node.value);
// 向右走一次
p = node.right;
}
}
}
midOrderN(root)
结果: