LeetCode刷题笔录 Binary Tree Inorder Traversal

最新推荐文章于 2022-06-30 15:29:15 发布

原创最新推荐文章于 2022-06-30 15:29:15 发布 · 785 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#dfs #leetcode #java #traversal #tree

LeetCode 同时被 3 个专栏收录

114 篇文章

订阅专栏

Java

50 篇文章

订阅专栏

Tree

15 篇文章

订阅专栏

本文介绍了一种二叉树中序遍历的迭代算法实现，该方法通过使用两个栈分别存储节点和计数，实现了无需递归即可完成中序遍历的过程。

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

Given a binary tree, return the inorder traversal of its nodes' values.

For example:
Given binary tree {1,#,2,3},

return [1,3,2].

Note: Recursive solution is trivial, could you do it iteratively?

这题和Preorder traversal的区别重点在于一个node第一次被从stack中pop出来时，不对其进行访问，而是要将其放回stack。在第二次pop时再进行访问。

在纸上画一画就可以得到以下要点：

1.需要记录一个node是第几次被从stack中pop

2.第一次pop出一个node后，按照right child, node, left child的顺序将这三个node（如果存在）放回stack

在这里我用了一个Stack<Integer> counterStack来记录每个node是第几次被pop出stack。初始值是0，第一次被pop后数值变为1. 两个stack的元素一一对应，这样就可以知道当前node的count。

public class Solution {
    public ArrayList<Integer> inorderTraversal(TreeNode root) {
        ArrayList<Integer> result = new ArrayList<Integer>();
        if(root == null)
            return result;
        Stack<TreeNode> nodeStack = new Stack<TreeNode>();
        Stack<Integer> counterStack = new Stack<Integer>();
        nodeStack.push(root);
        counterStack.push(0);
        
        while(!nodeStack.isEmpty()){
            TreeNode node = nodeStack.pop();
            int count = counterStack.pop();
            //if this is the second time that this node is popped out, visit it
            if(count == 1){
                result.add(node.val);
            }
            //else push the node's right child to the stack, itself back to the stack, add its counter, and adds its left child
            else{
                if(node.right != null){
                    nodeStack.push(node.right);
                    counterStack.push(0);
                }
                nodeStack.push(node);
                counterStack.push(1);
                if(node.left != null){
                    nodeStack.push(node.left);
                    counterStack.push(0);
                }
            }
        }
        
        return result;
    }
}