题目描述:
从上往下打印出二叉树的每个节点,同层节点从左至右打印。
思路:
每一次打印节点的时候,如果该节点有子节点,则把该节点的一个子节点放到一个队列的末尾,按照先左子节点再右子节点的顺序。接下来队列的头部取出最早进入队列的节点,重复前面的操作,直至队列中的元素都被打印出来。
代码:
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){//二叉树为空树
return list;
}
TreeNode p=null;
queue.offer(root);
while(!queue.isEmpty()){
p=queue.poll();//出队列的节点
list.add(p.val);
if(p.left!=null){
queue.offer(p.left);
}
if(p.right!=null){
queue.offer(p.right);
}
}
return list;
}
}