题目描述:
从上往下打印出二叉树的每个节点,同层节点从左至右打印。
解题思路:
题目实际上就是二叉树的层序遍历,如上图,我们借助队列来存储节点来实现层序遍历的效果,具体操作:
1. 先将根节点入队列
2. 将根节点出队列,打印;然后将其左右子树依次入队列(若其存在)
3. 重复步骤2,直到将树遍历完为止。
上面步骤最核心的就是步骤2,步骤2为什么能够实现层序的效果呢?当该节点出栈的时候,才会将下一层的节点(其左右子树)入栈,这样就保证了一层一层的打印效果;由于左子树一直在右子树前面进行操作,所以,不会乱序,可以实现从左往右的效果。
Java代码实现:
public ArrayList<Integer> PrintFromTopToBottom(TreeNode root) {
Queue<TreeNode> myQueue = new LinkedList<>();//注意Queue是一个接口
ArrayList<Integer> list = new ArrayList<>();
if (root==null)
return list;
myQueue.offer(root);//先将根节点入队列
while(!myQueue.isEmpty()){//一直将队列中所有元素都出队为止
TreeNode tempNode = myQueue.poll();//出队操作
list.add(tempNode.val);
if (tempNode.left!=null)//节点的左右子树入队操作
myQueue.offer(tempNode.left);
if (tempNode.right!=null)
myQueue.offer(tempNode.right);
}
return list;
}
END