说实话,有关二叉树遍历这块,特别是迭代版本,网上好多写的糊里糊涂的,尤其是将三种遍历统一风格的,基本都是看到一头雾水,我想了个比较直观点(自认为)
首先,以下图二叉树为例,
使用迭代法,无论哪种遍历顺序都要首先要开一个栈,同时还需要一个指针cur用于控制当前
接下来是各遍历的代码
前序遍历
let preorder=function(root){
if(!root) return null;
let stack=[],res=[],cur=root;
//stack.push(root);
while(cur||stack.length){
if(cur){
res.push(cur.val);
stack.push(cur);
cur=cur.left;
}
else{
cur=stack.pop();
cur=cur.right;
}
}
}
while循环的条件是一个或条件,当cur不为空时,此时cur一直向左走,因为是先序遍历,边走边将途径的节点值加入结果集,同时也要将当前的节点压入栈,用于之后的处理,如下图
在第一个向左深入过程中,先序遍历将cur指针节点的值存入res结果集中,同时把节点都压入栈中,最后cur变为null,但stack存有所有最左路径的节点,循环继续。
此时,while循环体内满足else条件,栈开始不断弹出,同时cur指向这些弹出的节点,并且一旦这些节点存在右孩子,cur就指向右孩子,相当于是进入一颗子树,上图最终cur指向根节点的右孩子节点7,然后循环体内第一个条件再次满足,继续之前的过程。直到结束./
中序遍历
中序遍历与前序遍历的不同仅仅在于添加结果集的时机不同,因为中序是先左再中的,所以在压入栈的时候不能读节点,也就是不能添加到结果集中,但在栈弹出的时候需要读取,代码为
let inorder=function(root){
if(!root) return null;
let stack=[],res=[],cur=root;
//stack.push(root);
while(cur||stack.length){
if(cur){
stack.push(cur);
cur=cur.left;
}
else{
cur=stack.pop();
res.push(cur.val);
cur=cur.right;
}
}
}
后序遍历
后序遍历的迭代版本,最易理解的方式普遍为,先改变先序遍历的根左右为根右左,然后逆序输出结果就得到后序遍历左右根,所以只需调换先序遍历的推入栈的节点为先右后左即可,代码为
let postorder=function(root){
if(!root) return null;
let stack=[],res=[],cur=root;
//stack.push(root);
while(cur||stack.length){
if(cur){
//console.log(cur.val);
res.push(cur.val);
stack.push(cur);
cur=cur.right;
}
else{
cur=stack.pop();
cur=cur.left;
}
}
console.log( res.reverse());
}
由此可见,三种遍历是可以写成一个风格的,网上普遍流程的三种只需调换两行代码的版本(代码随想录)虽然看上去牛逼,但逻辑很抽象,很复杂,不如这里调换读节点的时机来得直观,另外,后序虽然比代码随想录的略多一步翻转,但也直观很多。