[LeetCode] 94. 中序遍历二叉树

最新推荐文章于 2024-12-24 23:56:39 发布

原创最新推荐文章于 2024-12-24 23:56:39 发布 · 2.2k 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #算法 #二叉树

算法习题同时被 2 个专栏收录

18 篇文章

订阅专栏

算法题库

17 篇文章

订阅专栏

本文介绍了一种不使用递归实现二叉树中序遍历的方法。通过运用栈来辅助处理节点，遵循特定规则：将当前非空节点压栈、左子节点不空则继续压栈、左子节点为空则弹出节点记录值并将右子节点压栈，最终实现了中序遍历。

传送门

题目描述

给定一个二叉树，返回其中序遍历。

例如：
给定二叉树 [1,null,2,3],

返回 [1,3,2].

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

我的想法

题中说明了要求使用迭代法

使用一个栈来存储二叉树节点，根据中序遍历的规则，我们可以推算出这样的规律：
1. 将当前非空节点入栈
2. 如果左子节点不为空，则继续将左子节点入栈
3. 如果左子节点为空，则抛出栈顶节点并记录 val 值，然后将其右子节点入栈
4. 重复 1、2、3 步骤直至栈空

AC 代码：

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public List<Integer> inorderTraversal(TreeNode root) {
        Stack<TreeNode> stack = new Stack<>();
        List<Integer> list = new ArrayList<>();
        while(true){
            if (root != null){
                stack.push(root);
                root = root.left;
            } else {
                if (stack.empty()) {
                    return list;
                }
                root = stack.pop();
                list.add(root.val);
                root = root.right;
            }
        }
    }
}