LeetCode94. 二叉树的中序遍历

最新推荐文章于 2020-02-01 15:22:19 发布

原创最新推荐文章于 2020-02-01 15:22:19 发布 · 219 阅读

0 ·

CC 4.0 BY-SA版权

leecode 专栏收录该内容

4 篇文章

订阅专栏

博客围绕二叉树中序遍历展开，给出题目链接及要求，即给定二叉树返回中序遍历，还提出用迭代算法替代递归算法的进阶要求。介绍了递归和压栈两种解题思路，压栈思路包含节点左子树压栈、处理右子树等操作。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1.题目

https://leetcode-cn.com/problems/binary-tree-inorder-traversal/

给定一个二叉树，返回它的中序遍历。
示例:

输入: [1,null,2,3]
   1
    \
     2
    /
   3

输出: [1,3,2]

进阶: 递归算法很简单，你可以通过迭代算法完成吗？

2.思路

第一种方法是递归，
第二种方法是压栈：

将节点左子树全部压栈
弹出栈顶所有没有右子树的元素并将其加入遍历结果，直到出现右子树
若当前元素有右子树，则将当前元素加入遍历结果
指向当前节点右子树
若当前节点为空，则结束遍历；若当前节点不为空，则继续循环上述操作

3.解题：
方法一

public class InorderTraversal {
    public List<Integer> inorderTraversal(TreeNode root) {
        if (root == null) {
            return new ArrayList<>();
        }
        List<Integer> result = new ArrayList<>();
        result.addAll(inorderTraversal(root.left));
        result.add(root.val);
        result.addAll(inorderTraversal(root.right));
        return result;
    }
}

方法二

// 方法2
public class InorderTraversal1 {
    public List<Integer> inorderTraversal(TreeNode root) {
        Stack<TreeNode> treeNodeStack = new Stack<>();
        List<Integer> result = new ArrayList<>();
        TreeNode node = root;
        while (node != null) {
            while (node != null) {
                treeNodeStack.push(node);
                node = node.left;
            }
            while (treeNodeStack.size() > 0 && (node = treeNodeStack.pop()).right == null) {
                result.add(node.val);
            }
            // node不可能为null
            if (node.right != null) {
                result.add(node.val);
            }
            node = node.right;
        }
        return result;
    }
}