- 问题描述
请实现一个函数按照之字形打印二叉树,即第一行按照从左到右的顺序打印,第二层按照从右至左的顺序打印,第三行按照从左到右的顺序打印,其他行以此类推。
- 解决方案
代码如下:
import java.util.ArrayList;
import java.util.LinkedList;
/*
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) {
LinkedList<TreeNode> list = new LinkedList<>();
ArrayList<ArrayList<Integer> > res = new ArrayList<>();
if(pRoot == null)
return res;
boolean flag = true;
list.addLast(pRoot);
while(list.size() > 0){
ArrayList<Integer> temp = new ArrayList<>();
int s = list.size();
for(int i = 0;i < s;i ++){
TreeNode node = list.pollFirst();
if(node.left != null)
list.addLast(node.left);
if(node.right != null)
list.addLast(node.right);
if(flag)
temp.add(node.val);
else
temp.add(0,node.val);
}
res.add(temp);
flag = !flag;
}
return res;
}
}