【剑指 Offer 学习】【面试题 23： 从上往下打印二叉树】【思路】

题目：

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

 

思路：

• 创建一个动态数组用于存储节点值、方法的返回
• 参数为空时返回空数组
• 创建一个队列用于存储未遍历的数组，并将根结点入队
• 记录当前处理的结点并控制台输出
• 将根结点存入要返回的数组
• 将当前结点的左右节点依次保存到队列的结尾（FIFO，先入先出）

 

代码：

package 从上往下打印二叉树23;

import java.util.ArrayList;
import java.util.LinkedList;

/**
 * 从上往下打印出二叉树的每个节点，同层节点从左至右打印。
 * @author Administrator
 */
class TreeNode {
	int val = 0;
	TreeNode left = null;
	TreeNode right = null;

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

public class Demo {
	public ArrayList<Integer> PrintFromTopToBottom(TreeNode root) {
		ArrayList<Integer> list = new ArrayList<>();
		// 参数为空返回空数组
		if (root == null) {
			return list;
		}
		// 用于存储未遍历的元素，即当前结点的两个子结点
		LinkedList<TreeNode> queue = new LinkedList<>();
		// 将根结点入队
		queue.add(root);

		while (!queue.isEmpty()) {
			// 用于记录当前处理的结点，poll：获取并移除队列的头
			TreeNode treeNode = queue.poll();
			// 输出队列头元素的值
			System.out.print(treeNode.val + " ");

			// 将根结点存入要返回的数组中
			list.add(treeNode.val);

			if (treeNode.left != null) {
				// 将树的左子结点添加到此队列的结尾
				queue.addLast(treeNode.left);
			}
			if (treeNode.right != null) {
				// 将树的右子结点添加到此队列的结尾
				queue.addLast(treeNode.right);
			}
		}
		return list;
	}

	public static void main(String[] args) {
		Demo demo = new Demo();
		//      8
		//    /   \
		//   6     10
		//  / \   / \
		// 5   7 9  11
		TreeNode root = new TreeNode(8);
		root.left = new TreeNode(6);
		root.left.left = new TreeNode(5);
		root.left.right = new TreeNode(7);
		root.right = new TreeNode(10);
		root.right.left = new TreeNode(9);
		root.right.right = new TreeNode(11);
		demo.PrintFromTopToBottom(root);
		System.out.println("\n");
        //         1
        //        /
        //       3
        //      /
        //     5
        //    /
        //   7
        //  /
        // 9
		TreeNode root2 = new TreeNode(1);
		root2.left = new TreeNode(3);
		root2.left.left = new TreeNode(5);
		root2.left.left.left = new TreeNode(7);
		root2.left.left.left.left = new TreeNode(9);
		demo.PrintFromTopToBottom(root2);
		System.out.println("\n");
		
		// 0
        //  \
        //   2
        //    \
        //     4
        //      \
        //       6
        //        \
        //         8
		TreeNode root3 = new TreeNode(0);
		root3.left = new TreeNode(2);
		root3.left.left = new TreeNode(4);
		root3.left.left.left = new TreeNode(6);
		root3.left.left.left.left = new TreeNode(8);
		demo.PrintFromTopToBottom(root3);
	}
}