题目:从上往下打印出二叉树的每个节点,同层节点从左至右打印。
分析:题目的要求即层次遍历二叉树。可实现的方法是利用队列,每次把根节点先存入,然后依次输出该根节点,判断该根节点有无左右节点,如果有左右节点的话,依次存入队列。然后队列再输出一个节点,并将这个输出节点作为根节点,然后判断该根节点有无左右节点,如此循环下去。
代码:
public class Solution {
public ArrayList<Integer> PrintFromTopToBottom(TreeNode root) {
ArrayList<Integer> list = new ArrayList<Integer>();
if(root==null){return list;}
Queue<TreeNode> queue = new LinkedList<TreeNode>();
queue.offer(root);
while(!queue.isEmpty()){
TreeNode treenode = queue.poll();
if(treenode.left!=null){
queue.offer(treenode.left);
}
if(treenode.right!=null){
queue.offer(treenode.right);
}
list.add(treenode.val);
}
return list;
}
}
分析二:不利用队列的话,用ArrayList模拟一个队列来存储TreeNode
public class Solution {
public ArrayList<Integer> PrintFromTopToBottom(TreeNode root) {
ArrayList<Integer> list = new ArrayList<Integer>();
ArrayList<TreeNode> queue = new ArrayList<TreeNode>();
if(root==null){return list;}
queue.add(root);
while(queue.size()!=0){
TreeNode temp=queue.remove(0);//删除并返回
if(temp.left!=null){queue.add(temp.left);}
if(temp.right!=null){queue.add(temp.right);}
list.add(temp.val);
}
return list;
}
}