从上到下打印二叉树

题目一

不分行从上到下打印二叉树：从上到下打印出二叉树的每个节点，同一层的节点按照从左到右的顺序打印。

class BinaryTreeNode{
int value;
BinaryTreeNode left;
BinaryTreeNode right;
}

思路

使用队列来完成，打印的同时也在往里面加入。

代码

public void fun(BinaryTreeNode root){
if(root==null){
return;
}
Queue<BinaryTreeNode> queue = new LinkedList<>();
queue.add(root);
while(!queue.empty()){
BinaryTreeNode temp=queue.poll();
System.out.print(temp.value);
if(temp.left != null){ queue.add(tempNode.left); }
if(temp.right != null){ queue.add(tempNode.right);}

}
}

题目二

从上到下按层打印二叉树，同一层的节点按从左到右的顺序打印 ，每一层打印一行。

思路

同样是层序遍历，与上一题不同的是，此处要记录每层的节点个数，在每层打印结束后多打印一个回车符。

代码

public void fun(BinaryTreeNode root){
if(root==null){
return;
}
Queue<BinaryTreeNode> queue = new LinkedList<>();
queue.add(root);
while(!queue.empty()){
for(int size=queue.size();size()>0;size--){
BinaryTreeNode temp=queue.poll();
System.out.print(temp.value);
if(temp.left != null){ queue.add(tempNode.left); }
if(temp.right != null){ queue.add(tempNode.right);}
}

System.out.println();

}
}

题目三

请实现一个函数按照之字形打印二叉树，即第一行按照从左到右的顺序打印，第二层按照从右至左的顺序打印，第三行按照从左到右的顺序打印，其他行以此类推。

思路

打印之字形的二叉树，需要两个栈。当我们在打印某一行结点时，把下一层的子结点保存到相应的栈里。如果当前打印的是奇数层，则先保存左子结点再保存右子结点到一个栈里；如果当前打印的是偶数层，则先保存右子结点再保存左子结点到第二个栈里。

代码

public ArrayList<ArrayList<Integer>> Print(TreeNode pRoot) {
ArrayList<ArrayList<Integer>> result = new ArrayList<>();
if (pRoot == null) return result;
Stack<TreeNode> s1 = new Stack<>();
Stack<TreeNode> s2 = new Stack<>();
ArrayList<Integer> list = new ArrayList<>();
list.add(pRoot.val);
result.add(list);
s1.push(pRoot);

while (s1.isEmpty() || s2.isEmpty()){
if (s1.isEmpty()&&s2.isEmpty())
break;
ArrayList<Integer> tmp = new ArrayList<>();
if (s2.isEmpty()) {
while (!s1.isEmpty()) {
if (s1.peek().right != null) {
tmp.add(s1.peek().right.val);
s2.push(s1.peek().right);
}
if (s1.peek().left != null) {
tmp.add(s1.peek().left.val);
s2.push(s1.peek().left);
}
s1.pop();
}
}else {
while (!s2.isEmpty()) {
if (s2.peek().left != null) {
tmp.add(s2.peek().left.val);
s1.push(s2.peek().left);
}
if (s2.peek().right != null) {
tmp.add(s2.peek().right.val);
s1.push(s2.peek().right);
}
s2.pop();
}
}
if(tmp.size()>0){
result.add(tmp);
}
}
return result;
}