题目 : 请实现一个函数按照之字形打印二叉树,即第一行按照从左到右的顺序打印,第二层按照从右至左的顺序打印,第三行按照从左到右的顺序打印,其他行以此类推。
思路 : 1.使用两个栈S1和S2存储奇数层和偶数层节点
2.奇数层从左到右push进左右子树(出队时就可以是先右后左了)
3.偶数层从右到左push进左右子树(出队时就可以是先左后右)
public static ArrayList<ArrayList<Integer>> print(TreeNode pRoot) {
ArrayList<ArrayList<Integer>> res = new ArrayList<>();
if (pRoot == null) {
return res;
}
Stack<TreeNode> stack1 = new Stack<>();
Stack<TreeNode> stack2 = new Stack<>();
int deepth = 1;
stack1.push(pRoot);
while (!stack1.isEmpty() || !stack2.isEmpty()) {
// 奇数层
if (deepth % 2 != 0) {
ArrayList<Integer> list1 = new ArrayList<>();//用来暂存结果
while (!stack1.isEmpty()) {
TreeNode cur = stack1.pop();
list1.add(cur.value);
if (cur.left != null) {
stack2.push(cur.left);
}
if (cur.right != null) {
stack2.push(cur.right);
}
}
if (!list1.isEmpty()) {
res.add(list1);
deepth++;
}
}
// 偶数层
else {
ArrayList<Integer> list2 = new ArrayList<>();//用来暂存结果
while (!stack2.isEmpty()) {
TreeNode cur = stack2.pop();
list2.add(cur.value);
if (cur.right != null) {
stack1.push(cur.right);
}
if (cur.left != null) {
stack1.push(cur.left);
}
}
if (!list2.isEmpty()) {
res.add(list2);
deepth++;
}
}
}
return res;
}