剑指offer | 把二叉树打印成多行

把二叉树打印成多行

从上到下按层打印成多行

分析:
使用队列

public class PrintTreeInLines<Key extends Comparable<Key>> {

    private class Node {
        public Key key;
        public Node left, right;

        public Node(Key key) {
            this.key = key;
        }
    }

    public void print(Node node) {
        if (node == null) {
            return;
        }

        Queue<Node> queue = new LinkedList<>();
        queue.add(node);
        int nextLevel = 0; // 下一层结点数
        int printed = 1; // 当前层没有打印的结点数
        while (!queue.isEmpty()) {
            Node current = queue.peek();
            System.out.printf("%d\t", current.key);

            if (current.left != null) {
                queue.add(current.left);
                ++nextLevel;
            }

            if (current.right != null) {
                queue.add(current.right);
                ++nextLevel;
            }

            queue.remove();
            --printed;
            if (printed == 0) {
                System.out.println("");
                printed =  nextLevel;
                nextLevel = 0;
            }
        }
    }
}
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

相关阅读更多精彩内容

友情链接更多精彩内容