二叉树的遍历-中序遍历-递归版本

解题思路分析

参考了邓俊辉老师的教材。
中序遍历:访问左子树、访问当前节点、访问右子树。
中序遍历的顺序如下图所示
在这里插入图片描述
可以发现对于每一个子树的中序遍历,总是先找到最左侧通路最下面的节点。然后访问它的右孩子,最后向上回溯。因此将中序遍历的顺序进一步抽象,如下图所示:
在这里插入图片描述
如上图所示,图中最左侧通路一共有L0 L1 ....Ldd+1个节点,因此中序遍历可以分为d+1端。每一段都可以由最左侧通路上的节点和其对应的右子树组成。当我们访问完一个节点和他对应的右子树后,我们需要回溯到最左侧通路上的上一个节点。例如访问完了Ld,就访问对应的右子树Td,这一段访问完之后,需要回溯到Ld-1,因此需要用栈来帮助这种回溯。

中序遍历可以总结为:

  1. 对于以root为根的二叉树,首先沿着最左侧通路行走(go,不做访问),并将沿途的节点压入栈中
  2. 第1步完成之后,节点肯定为空,此时如果栈为空,说明这个二叉树里面没有节点,那就可以返回了
  3. 如果栈不为空,就需要将栈顶元素topNode弹出,这个栈顶元素就是刚刚走过的最左侧通路的最下面的节点,根据上面的分析,我们需要访问这个节点,并以中序遍历的方式访问它的右子树,就又回到了第1步,也就是将1,2,3步套在一个循环里,将topNode赋值给1中的root
以中序遍历的方式访问以root为根的二叉树
定义一个node,初始化为root
while (true) {
	1、以node为根的二叉树,沿着最左侧通路行走,并将沿途的节点压入栈中;
	2、如果栈为空,break;
	3、如果栈不为空,弹出栈顶元素topNode,访问栈顶元素,将node = topNode.right
}

代码实现

// 中序遍历,迭代版本
public class Solution02 {
    public List<Integer> inorderTraversal(TreeNode root) {
        List<Integer> result = new ArrayList<>();
        Deque<TreeNode> stack = new ArrayDeque<>();
        TreeNode node = root;
        while (true) {
            goAloneLeft(node, stack);
            if (stack.isEmpty()) {
                break;
            }
            TreeNode topNode = stack.removeLast();
            result.add(topNode.val);
            node = topNode.right;
        }
        return result;
    }

    // 以node为根的子树,沿着最侧通路行走,将沿途的节点入栈,直到节点为空
    private void goAloneLeft(TreeNode node, Deque<TreeNode> stack) {
        while(node != null) {
            stack.addLast(node);
            node = node.left;
        }
    }
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值