树的中序遍历 (非递归) (Java)

最新推荐文章于 2021-11-23 21:15:06 发布

原创最新推荐文章于 2021-11-23 21:15:06 发布 · 432 阅读

0 ·

CC 4.0 BY-SA版权

8.4 数据结构&算法专栏收录该内容

15 篇文章

订阅专栏

本文介绍了一种使用非递归方式实现的二叉树中序遍历算法，通过构建特定的二叉树结构，并利用栈和哈希映射来追踪节点的状态，实现了对树的有序遍历。文章提供了完整的Java代码示例，展示了如何从根节点开始，依次访问左子树、根节点和右子树，最终输出节点的ID，得到中序遍历的结果。

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

一、问题

              1
       2             3
         4         5  6
                 7

输出该树中序遍历的结果

二、代码

Java

@Data
public class Node {
    private Integer id;
    private Node left;
    private Node right;

    public Node(Integer id) {
        this.id = id;
    }

    public Node(Integer id, Node left, Node right) {
        this.id = id;
        this.left = left;
        this.right = right;
    }
}




@Data
public class Tree {
    private Node root;

    /**
     * 建树 【V80】
     *
     *              1
     *       2             3
     *         4         5  6
     *                 7
     */
    public void build() {
        Node node7 = new Node(7);
        Node node6 = new Node(6);
        Node node5 = new Node(5, node7, null);
        Node node4 = new Node(4);
        Node node3 = new Node(3, node5, node6);
        Node node2 = new Node(2, null, node4);
        root = new Node(1, node2, node3);
    }

    /**
     * 中序遍历（非递归）
     */
    public void middleOrder(Node root) {
        if (root == null) {
            return;
        }

        if (root == null) {
            return;
        }

        //栈里所有的节点中：如果有左右孩子，其左右孩子已经入栈的节点
        Map<Integer, Boolean> childHasEnHeapMap = new HashMap<>();

        Deque<Node> heap = new LinkedList<>();
        heap.offerLast(root);
        while (!heap.isEmpty()) {
            Node top = heap.pollLast();
            Boolean childHasEnHeap = childHasEnHeapMap.get(top.getId());

            if (childHasEnHeap == null) {
                if (top.getRight() != null) {
                    heap.offerLast(top.getRight());
                }
                heap.offerLast(top);
                if (top.getLeft() != null) {
                    heap.offerLast(top.getLeft());
                }

                //左右孩子已入栈
                childHasEnHeapMap.put(top.getId(), Boolean.TRUE);
            } else {
                childHasEnHeapMap.remove(top.getId());
                System.out.print(top.getId() + "   ");
            }
        }
    }



    public static void main(String[] args) {
        Tree tree = new Tree();
        tree.build();

        tree.middleOrder(tree.getRoot());
    }
}

运行结果

2 4 1 7 5 3 6