1.本题知识点
二叉树,栈
2. 题目描述
请实现一个函数按照之字形打印二叉树,即第一行按照从左到右的顺序打印,第二层按照从右至左的顺序打印,第三行按照从左到右的顺序打印,其他行以此类推。
3. 思路
关键点: 用两个栈分别保存奇数层和偶数层的节点。
如上图,打印顺序及过程如下:
初始化,将1放入stack1.
1 奇数层 ---- stack1弹出1打印,并将1的左右子节点放入stack2.
3 2 偶数层 ---- stack2弹出2,3,并将左右子节点放入stack1.
4 5 6 7 奇数层 ---- stack1弹出,并将左右子节点放入stack2.
15 14 13 12 12 10 9 8 偶数层 ---- stack2弹出,…
Java版本:
import java.util.ArrayList;
import java.util.Stack;
/*
public class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
*/
public class Solution {
public ArrayList<ArrayList<Integer> > Print(TreeNode pRoot) {
ArrayList<ArrayList<Integer>> result = new ArrayList<>();//final
Stack<TreeNode> s1 = new Stack<>();//奇数层栈
s1.push(pRoot);//初始化
Stack<TreeNode> s2 = new Stack<>();//偶数层栈
int layer = 1; //记录奇数偶数的标识
while(!s1.isEmpty() || !s2.isEmpty()){
//保存每层节点数据
ArrayList<Integer> layerList = new ArrayList<>();
//如果是奇数层
if(layer % 2 != 0){
//遍历奇数栈中每一个节点,打印同时将其子节点放入偶数栈中
while(!s1.isEmpty()){
TreeNode node = s1.pop();
if(node != null){
layerList.add(node.val);//打印
//将左右子树加入偶数栈
s2.push(node.left);
s2.push(node.right);
}
}
//当前层放入最终列表中
if(!layerList.isEmpty()){
result.add(layerList);
}
layer++;//处理完一层后,改变奇偶标识符
}
//如果是偶数层
else{
//遍历偶数栈中每一个节点,打印同时将其子节点放入奇数栈中
while(!s2.isEmpty()){
TreeNode node = s2.pop();
if(node != null){
layerList.add(node.val);//打印
//将左右子树加入奇数栈
//由于奇数栈打印是从左往右,所以奇数栈入栈要从右往左
s1.push(node.right);
s1.push(node.left);
}
}
//当前层放入最终列表中
if(!layerList.isEmpty()){
result.add(layerList);
}
layer++;//处理完一层后,改变奇偶标识符
}
}
return result;
}
}