剑指Offer—二叉树分层打印
解题思路:建立一个队列,存放结点,创建两个指针指向最近结点和行尾结点
/**
* 分层遍历二叉树(使用队列实现)
* 1、新建队列,且创建两个临时节点元素last和nlast,初始值均为头结点,栈顶元素压入队列
* 2、循环遍历队列,如果不为空,队头元素出队赋值给node,并打印
* 如果队头元素存在左孩子,左孩子入队,左孩子赋值给nlast。
* 如果队头元素存在右孩子,右孩子入队,右孩子赋值给nlast
* 如果队头元素等于last,换行并打印行号,把nlast赋给last.
* 3、直至队列为空
* @param root 头结点
*/
public ArrayList<Integer> PrintFromTopToBottom(TreeNode root) {
ArrayList<Integer> rs = new ArrayList<>();
LinkedList<TreeNode> queue = new LinkedList<>();
TreeNode node = null;
TreeNode last = root,nlast = root;
if (root!=null){
queue.offer(root);
}
while (!queue.isEmpty()){
node = queue.poll();
rs.add(node.val);
if (node.left!=null){
queue.offer(node.left);
nlast = node.left;
}
if (node.right!=null){
queue.offer(node.right);
nlast = node.right;
}
if (node == last){
last = nlast;
}
}
return rs;
}