树的中序遍历新思路

最新推荐文章于 2023-12-20 13:33:39 发布

Fire king

最新推荐文章于 2023-12-20 13:33:39 发布

阅读量66

点赞数 5

分类专栏： LeetCode 文章标签：二叉树中序遍历栈

本文链接：https://blog.csdn.net/m0_48333563/article/details/124801432

版权

LeetCode 专栏收录该内容

18 篇文章 0 订阅

订阅专栏

这个博客介绍了一个利用颜色标记实现的中序遍历二叉树的方法。通过创建`ColorNode`类来存储节点及其颜色（白色或灰色），然后使用栈进行辅助遍历。在遍历过程中，首先将根节点标记为白色并压栈，接着在栈不为空的情况下，弹出节点并根据颜色决定是继续压入其子节点还是将其值添加到结果列表中。

摘要由CSDN通过智能技术生成

class Solution {
    class ColorNode {
        TreeNode node;
        String color;
        
        public ColorNode(TreeNode node,String color){
            this.node = node;
            this.color = color;
        }
    }
    public List<Integer> inorderTraversal(TreeNode root) {
        if(root == null) return new ArrayList<Integer>();
            
        List<Integer> res = new ArrayList<>();
        Stack<ColorNode> stack = new Stack<>();
        stack.push(new ColorNode(root,"white"));
        
        while(!stack.empty()){
            ColorNode cn = stack.pop();
            
            if(cn.color.equals("white")){
                if(cn.node.right != null) stack.push(new ColorNode(cn.node.right,"white"));
                stack.push(new ColorNode(cn.node,"gray"));
                if(cn.node.left != null)stack.push(new ColorNode(cn.node.left,"white"));
            }else{
                res.add(cn.node.val);
            }
        }
        
        return res;
    }
}