题目描述
从上往下打印出二叉树的每个节点,同层节点从左至右打印。
解题思路
利用一个队列,如果有子节点就将子节点入队,出队的时候将子节点入队
import java.util.ArrayList;
import java.util.LinkedList;
import java.util.Queue;
/**
public class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
*/
public class Solution {
public ArrayList<Integer> PrintFromTopToBottom(TreeNode root) {
ArrayList<Integer> list=new ArrayList<Integer>(); //存放结果
Queue<TreeNode> queue= new LinkedList<TreeNode>(); //辅助队列
if (root!=null){
//根节点入队
queue.offer(root);
}
//队列不为空,执行循环
while (!queue.isEmpty()){
TreeNode node=queue.poll();
list.add(node.val); //将队列元素输出
//如果有左节点,就把左节点加入
if (node.left!=null){
queue.offer(node.left);
}
//如果有右节点,就把右节点加入
if(node.right!=null){
queue.offer(node.right);
}
}
return list;
}
}