一、题目描述
请实现一个函数按照之字形打印二叉树,即第一行按照从左到右的顺序打印,第二层按照从右至左的顺序打印,第三行按照从左到右的顺序打印,其他行以此类推。
二、解题思路
使用两个栈stack1,stack2。打印某一行结点时,把下一层的子结点保存到另一栈里:
a. 如果当前打印的是stack1出栈的奇数层结点,则先保存左子结点再保存右子结点到栈stack2里;
b. 如果当前打印的是stack2出栈的偶数层结点,则先保存右子结点再保存左子结点到栈stack1里。
例如:
[01]
[02] [03]
[04] [05] [06] [07]
[08] [09] [10] [11] [12] [13] [14] [15]
将结点1(第一层)存入stack1,layer=1
stack1:{1}
stack2:{}
list:{}
layer:1
layer为奇数,出栈stack1所有结点,并将所有结点的子节点从左至右入栈stack2,layer+1
stack1:{}
stack2:{2,3}
list:{[1]}
layer:2
layer为偶数,出栈stack2所有结点,并将所有结点的子节点从右至左入栈stack1,layer+1
stack1:{7,6,5,4}
stack2:{}
list:{[1],[3,2]}
layer:3
layer为奇数,出栈stack1所有结点,并将所有结点的子节点从左至右入栈stack2,layer+1
stack1:{}
stack2:{8,9,10,11,12,13,14,15}
list:{[1],[3,2],[4,5,6,7]}
layer:4
layer为偶数,出栈stack2所有结点,并将所有结点的子节点从右至左入栈stack1,layer+1
stack1:{}
stack2:{}
list:{[1],[3,2],[4,5,6,7],[15,14,13,12,11,10,9,8]}
layer:5
三、编程实现
import java.util.ArrayList;
import java.util.Stack;
public class Solution {
public ArrayList<ArrayList<Integer>> Print(TreeNode pRoot) {
int layer = 1;
Stack<TreeNode> s1 = new Stack<>();
Stack<TreeNode> s2 = new Stack<>();
// 先将根节点(第一层)入栈s1
s1.push(pRoot);
ArrayList<ArrayList<Integer>> list = new ArrayList<>();
while (!s1.isEmpty() || !s2.isEmpty()) {
// 奇数层
if (layer % 2 != 0) {
ArrayList<Integer> temp = new ArrayList<>();
while (!s1.isEmpty()) {
// 出栈所有结点
TreeNode node = s1.pop();
if (node != null) {
temp.add(node.val);
// 将所有结点的左右子节点从左至右入栈s2
s2.push(node.left);
s2.push(node.right);
}
}
if (!temp.isEmpty()) {
list.add(temp);
layer++;
}
} else {
// 偶数层
ArrayList<Integer> temp = new ArrayList<>();
while (!s2.isEmpty()) {
// 出栈所有结点
TreeNode node = s2.pop();
if (node != null) {
temp.add(node.val);
// 将所有结点的左右子节点从右至左入栈s2
s1.push(node.right);
s1.push(node.left);
}
}
if (!temp.isEmpty()) {
list.add(temp);
layer++;
}
}
}
return list;
}
}