演算法與資料結構面試題(20)-層序遍曆二叉樹__JAVA

題目

輸入一顆二元樹，從上往下按層列印樹的每個結點，同一層中按照從左往右的順序列印。

例如下面的二叉樹，列印的結果是：8，6，10，5，7，9，11

解題思路

利用BFS（寬度優先）遍曆二叉樹。直接上代碼

Queue<TreeNode> queue = new LinkedList<TreeNode>();      /**     * 層序遍曆二叉樹層數從上到小,同一層從左至右     */    public void layerOrder(TreeNode root) {        if (root == null)            return;        queue.add(root);        while (!queue.isEmpty()) {            TreeNode node = queue.poll();            if (node == null)                continue;            if (node.lchild != null) {                queue.add(node.lchild);            }            if (node.rchild != null) {                queue.add(node.rchild);            }            System.out.println(node.getData());        }    }

用隊列的形式儲存待遍曆的結點。遍曆的時候先儲存左右結點，然後列印資訊

public static void main(String[] args) {TreeNode leaf1 = new TreeNode("5", null, null); // 沒有左右孩子節點TreeNode leaf2 = new TreeNode("7", null, null); // 沒有左右孩子節點TreeNode leaf3 = new TreeNode("9", null, null); // 沒有左右孩子節點TreeNode leaf4 = new TreeNode("11", null, null); // 沒有左右孩子節點TreeNode n3 = new TreeNode("6", leaf1, leaf2); // 根節點左子樹TreeNode n4 = new TreeNode("10", leaf3, leaf4); // 根節點右子樹TreeNode root = new TreeNode("8", n3, n4); // 建立根節點BinaryTree bt = new BinaryTree(root);bt.layerOrder(root);}

輸出結果

861057911