原题链接:把二叉树打印成多行_牛客题霸_牛客网
描述:
给定一个节点数为 n 二叉树,要求从上到下按层打印二叉树的 val 值,同一层结点从左至右输出,每一层输出一行,将输出的结果存放到一个二维数组中返回。
例如:
给定的二叉树是{1,2,3,#,#,4,5}
该二叉树多行打印层序遍历的结果是
[
[1],
[2,3],
[4,5]
]
数据范围:二叉树的节点数 0≤n≤1000,0≤val≤1000
要求:空间复杂度 O(n),时间复杂度 O(n)
解题思路:
也是一层一层的遍历
代码:
public ArrayList<ArrayList<Integer>> Print(TreeNode pRoot) {
ArrayList<ArrayList<Integer>> list2 = new ArrayList<>();
if (pRoot == null) {
return list2;
}
ArrayList<TreeNode> inList = new ArrayList<>();
inList.add(pRoot);
ArrayList<TreeNode> outList = new ArrayList<>();
while (inList.size() > 0) {
outList.clear();
ArrayList<Integer> integers = PrintLevel(inList, outList);
inList.clear();
inList.addAll(outList);
list2.add(integers);
}
return list2;
}
public ArrayList<Integer> PrintLevel(ArrayList<TreeNode> inList, ArrayList<TreeNode> outList) {
ArrayList<Integer> list = new ArrayList<>();
for (int i = 0; i < inList.size(); i++) {
TreeNode node = inList.get(i);
list.add(node.val);
if (node.left != null) {
outList.add(node.left);
}
if (node.right != null) {
outList.add(node.right);
}
}
return list;
}