BFS实现从上往下打印二叉树

【推荐】2019 Java 开发者跳槽指南.pdf(吐血整理) >>>

题目描述

从上往下打印出二叉树的每个节点，同层节点从左至右打印

深度优先搜索算法（Breadth-First-Search）

深度优先算法是一种图形搜索算法。简单的说，BFS是从根节点开始，沿着树(图)的宽度遍历树(图)的节点。如果所有节点均被访问，则算法中止。一般用队列数据结构来辅助实现BFS算法。

算法步骤

1. 首先将根节点放入队列中。
2. 从队列中取出第一个节点，并检验它是否为目标。
如果找到目标，则结束搜寻并回传结果。
否则将它所有尚未检验过的直接子节点加入队列中。
3. 若队列为空，表示整张图都检查过了——亦即图中没有欲搜寻的目标。结束搜寻并回传“找不到目标”。
4. 重复步骤2

代码实现

public ArrayList<Integer> printFromTopToBottom(TreeNode root) {
		if (root == null)
			return new ArrayList<Integer>();
		ArrayList<Integer> list = new ArrayList<Integer>();
		ArrayDeque<TreeNode> queue = new ArrayDeque<TreeNode>();
		// 首先将根节点放入队列中
		queue.offer(root);
		while (!queue.isEmpty()) {// 队列不空,还有节点未搜索
			// 根节点出队
			TreeNode current = queue.poll();
			// 将根节点放入结果集中
			list.add(current.val);
			if (current.left != null) {// 左节点不空,先搜索左节点
				// 先将左节点放入队列
				queue.offer(current.left);
			}
			if (current.right != null) {// 右节点不空,搜索右节点
				// 再将右节点放入队列
				queue.offer(current.right);
			}
		}
		return list;
	}

	public class TreeNode {
		int val = 0;
		TreeNode left = null;
		TreeNode right = null;

		public TreeNode(int val) {
			this.val = val;
		}
	}

来源：oschina

链接：https://my.oschina.net/u/2503731/blog/660865