前中后序迭代统一格式遍历法(最好理解)js版本

说实话,有关二叉树遍历这块,特别是迭代版本,网上好多写的糊里糊涂的,尤其是将三种遍历统一风格的,基本都是看到一头雾水,我想了个比较直观点(自认为)
首先,以下图二叉树为例,在这里插入图片描述
使用迭代法,无论哪种遍历顺序都要首先要开一个栈,同时还需要一个指针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());
}

由此可见,三种遍历是可以写成一个风格的,网上普遍流程的三种只需调换两行代码的版本(代码随想录)虽然看上去牛逼,但逻辑很抽象,很复杂,不如这里调换读节点的时机来得直观,另外,后序虽然比代码随想录的略多一步翻转,但也直观很多。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值