【剑指Offer】从上往下打印二叉树

题目描述

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

思路

1.从上到下，从左到右打印，这是一个遍历二叉树的问题，能够用队列的先进先出的特点来满足

2.开辟一个队列，先让根节点入队，循环判断队列是否为空，若不空，则队头元素出队并将其val值放入list中，若其左右孩子节点不空，则将左右孩子节点入队

3.返回最后得到的list集合

代码实现（Java）

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

public class Solution {
    public ArrayList<Integer> PrintFromTopToBottom(TreeNode root) {
		ArrayList<Integer> list = new ArrayList<Integer>();
		//鲁棒性(当树为空时，返回一个空的列表，这里不要写成return null; 提交会报空指针
		if(root == null)
			return list;
		//list用于存储最终的序列，Queue辅助二叉树的遍历
		Queue<TreeNode> queue = new LinkedList<TreeNode>();
		//先在queue中放入根节点元素
		queue.offer(root);
		while(!queue.isEmpty()) {
			TreeNode temp = queue.poll();//queue.remove()，也能实现移除并返回队头元素，但是后者在queue为空时为抛异常，前者返回null
			list.add(temp.val);
			if(temp.left != null)
				queue.offer(temp.left);
			if(temp.right != null)
				queue.offer(temp.right);
		}
		return list;
	}
}

测试输出

构建一个节点的左右孩子节点，测试上述二叉树的输出

	public void setLeftAndRight(TreeNode root, TreeNode left, TreeNode right) {
		root.left = left;
		root.right = right;
	}
	

	@Test
	public void test() {
		TreeNode node1 = new TreeNode(5);
		TreeNode node2 = new TreeNode(2);
		TreeNode node3 = new TreeNode(4);
		TreeNode node4 = new TreeNode(3);
		TreeNode node5 = new TreeNode(6);
		TreeNode node6 = new TreeNode(7);
		TreeNode node7 = new TreeNode(8);
		
		setLeftAndRight(node1, node2, node3);
		setLeftAndRight(node2, node4, null);
		setLeftAndRight(node3, node5, node6);
		setLeftAndRight(node4, null, node7);
		
		ArrayList<Integer> list = PrintFromTopToBottom(node1);
		System.out.println(list);
	}