题目:
分行从上到下打印二叉树 从上到下打印出二叉树的每个节点,同一层的节点按照从左到右的顺序打印。
思路:
该题即为对二叉树的层序遍历,结点满足先进先出的原则,采用队列。每从队列中取出头部结点并打印,若其有子结点,把子结点放入队列尾部,直到所有结点打印完毕。
根据当前层结点数目来打印当前层结点,同时计算下一层结点数目,之后令pCount等于nextCount,重复循环,直到打印完毕。
package jianZhiOffer;
import java.util.LinkedList;
public class Demo3201 {
public static void main(String[] args) {
ListNode root = new ListNode(8);
root.left = new ListNode(6);
root.right = new ListNode(10);
root.left.left = new ListNode(5);
root.left.right = new ListNode(7);
root.right.left = new ListNode(9);
root.right.right = new ListNode(11);
printTree2(root);
}
public static void printTree2(ListNode root) {
if (root == null)
return;
LinkedList<ListNode> queue = new LinkedList<ListNode>(); //队列
queue.offer(root);
ListNode node = null;
int pCount = 0; //当前层结点数目
int nextCount = 1; //下一层结点数目
while (!queue.isEmpty()) {
pCount = nextCount;
nextCount = 0;
//打印当前层数字,并计算下一层结点数目
for (int i = 1; i <= pCount; i++) {
node = queue.poll();
System.out.print(node.val + " ");
if (node.left != null) {
queue.offer(node.left);
nextCount++;
}
if (node.right != null) {
queue.offer(node.right);
nextCount++;
}
}
System.out.println();
}
}
}
之字形打印二叉树
题目描述
请实现一个函数按照之字形打印二叉树,即第一行按照从左到右的顺序打印,第二层按照从右至左的顺序打印,第三行按照从左到右的顺序打印,其他行以此类推。
思路
1.使用两个栈来分别存储奇数层节点和偶数层节点。
2.注意两个栈的插入顺序是不同的。
3.对于奇数层来说,也就是从左往右的顺序,先添加左子树,然后添加右子树。对于偶数层,刚好相反,先添加右子树,然后添加左子树。
package jianZhiOffer;
import java.util.ArrayList;
import java.util.LinkedList;
import java.util.Stack;
/*
* 题目三:之字形打印二叉树
* 请实现一个函数按照之字形顺序打印二叉树,即第一行按照从左到右得顺序打印,第二层按照从右到左得顺序打印,
* 第三行再按照从左到右的顺序打印,其他以此类推。
*/
public class Demo3202 {
public static void main(String[] args) {
ListNode root = new ListNode(8);
root.left = new ListNode(6);
root.right = new ListNode(10);
root.left.left = new ListNode(5);
root.left.right = new ListNode(7);
root.right.left = new ListNode(9);
root.right.right = new ListNode(11);
System.out.println(PrintTree(root));
}
public static ArrayList<ArrayList<Integer>> PrintTree(ListNode pRoot){
int level = 1;
//s1存奇数层节点
Stack<ListNode> s1 = new Stack<ListNode>();
s1.push(pRoot);
//s2存偶数层节点
Stack<ListNode> s2 = new Stack<ListNode>();
ArrayList<ArrayList<Integer>> list = new ArrayList<ArrayList<Integer>>();//集合中放集合
while(!s1.empty() || !s2.empty()) {
if(level%2 != 0) { //奇数层
ArrayList<Integer> temp = new ArrayList<Integer>();
while(!s1.empty()) {
ListNode node = s1.pop();
if(node != null) {
temp.add(node.val);
s2.push(node.left);
s2.push(node.right);
}
}
if(!temp.isEmpty()) {
list.add(temp);
level++;
}
}else {
ArrayList<Integer> temp = new ArrayList<Integer>();
while(!s2.empty()) {
ListNode node = s2.pop();
if(node != null) {
temp.add(node.val);
s1.push(node.right);
s1.push(node.left);
}
}
if(!temp.isEmpty()) {
list.add(temp);
level++;
}
}
}
return list;
}
}