算法与数据结构面试题(20)-层序遍历二叉树

题目

输入一颗二元树，从上往下按层打印树的每个结点，同一层中按照从左往右的顺序打印。

例如下面的二叉树，打印的结果是：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);
}

输出结果

8
6
10
5
7
9
11