题目描述
从上到下按层打印二叉树,同一层结点从左至右输出。每一层输出一行。
题解
大体框架与层次遍历相同,只是为了记录每层的节点数目,设置一个变量size,用于对当前层的节点进行遍历。
ArrayList<ArrayList<Integer>> Print(TreeNode root) {
ArrayList<ArrayList<Integer>> res = new ArrayList<>();
Queue<TreeNode> queue = new LinkedList<>();
queue.offer(root);
if (root == null)
return res;
while (!queue.isEmpty()) {
int size = queue.size();
ArrayList<Integer> list = new ArrayList<>();
for (int i = 0; i < size; i++) {
TreeNode temp = queue.poll();
list.add(temp.val);
if (temp.left != null)
queue.offer(temp.left);
if (temp.right != null)
queue.offer(temp.right);
}
res.add(list);
}
return res;
}
网友评论