二叉樹的層次遍歷（按層換行打印）

概述

二叉樹的層次遍歷只需使用一個隊列即可，但若需要按層來換行打印則稍麻煩一些。

思路

增加兩個TreeNode：last和nlast
last：表示當前遍歷層最右結點
nlast：表示下一層最右結點
遍歷時，每次將nlast指向插入隊列元素，最后一個插入結點時即最右結點。插入左右孩子之后，檢測last是否為當前輸出結點，若是，則表示需要進行換行，並將last指向下一層的nlast。

代碼

/*
public class TreeNode {
    int data;
    TreeNode left;
    TreeNode right;

    public TreeNode(int data) {
        this.data = data;
    }
}
*/
public class Tree {
    TreeNode last;
    TreeNode nlast;
    public void printTree(TreeNode root) {
        Queue<TreeNode> queue = new LinkedList<>();
        queue.add(root);
        last = root;
        nlast = root;
        while (!queue.isEmpty()) {
            TreeNode t = queue.peek();
            System.out.print(queue.poll().data + " ");
            if (t.left != null) {
                queue.add(t.left);
                nlast = t.left;
            }
            if (t.right != null) {
                queue.add(t.right);
                nlast = t.right;
            }
            // 如果當前輸出結點是最右結點，那么換行
            if (last == t) {
                System.out.println();
                last = nlast;
            }
        }
    }

    public static void main(String[] args) {
        // 構建二叉樹
        TreeNode root = new TreeNode(1);
        root.left = new TreeNode(2);
        root.right = new TreeNode(3);
        root.left.left = new TreeNode(4);
        root.right.left = new TreeNode(5);
        root.right.right = new TreeNode(6);
        root.right.left.left = new TreeNode(7);
        root.right.left.right = new TreeNode(8);
        Tree test = new Tree();
        test.printTree(root);
    }
}