原题链接:按之字形顺序打印二叉树_牛客题霸_牛客网
描述:
给定一个二叉树,返回该二叉树的之字形层序遍历,(第一层从左向右,下一层从右向左,一直这样交替)
数据范围:0≤n≤1500,树上每个节点的val满足 |val| <= 100∣val∣<=100
要求:空间复杂度:O(n)O(n),时间复杂度:O(n)O(n)
例如:
给定的二叉树是{1,2,3,#,#,4,5}
该二叉树之字形层序遍历的结果是
[
[1],
[3,2],
[4,5]
]
示例1
输入:
{1,2,3,#,#,4,5}
返回值:
[[1],[3,2],[4,5]]
说明:
如题面解释,第一层是根节点,从左到右打印结果,第二层从右到左,第三层从左到右。
解题思路:
一层一层的遍历
代码:
public ArrayList<ArrayList<Integer>> Print(TreeNode pRoot) {
ArrayList<ArrayList<Integer>> list = new ArrayList<>();
if (pRoot == null) {
return list;
}
ArrayList<TreeNode> inList = new ArrayList<>();
inList.add(pRoot);
ArrayList<TreeNode> outList = new ArrayList<>();
boolean isFromLeft = true;
while (inList.size() > 0) {
outList.clear();
ArrayList<Integer> integers = PrintLevel(isFromLeft, inList, outList);
inList.clear();
inList.addAll(outList);
list.add(integers);
isFromLeft = !isFromLeft;
}
return list;
}
public ArrayList<Integer> PrintLevel(boolean isFromLeft, ArrayList<TreeNode> inList, ArrayList<TreeNode> outList) {
ArrayList<Integer> list = new ArrayList<>();
for (int i = 0; i < inList.size(); i++) {
TreeNode node = inList.get(i);
if (isFromLeft) {
list.add(node.val);
} else {
list.add(0, node.val);
}
if (node.left != null) {
outList.add(node.left);
}
if (node.right != null) {
outList.add(node.right);
}
}
return list;
}