返回层次数二叉树java,leetcode树之从上到下打印二叉树

序

本文主要记录一下leetcode树之上到下打印二叉树

题目

从上到下按层打印二叉树，同一层的节点按从左到右的顺序打印，每一层打印到一行。

例如:

给定二叉树: [3,9,20,null,null,15,7],

3

/ \

9 20

/ \

15 7

返回其层次遍历结果：

[

[3],

[9,20],

[15,7]

]

提示：

节点总数 <= 1000

注意：本题与主站 102 题相同：https://leetcode-cn.com/problems/binary-tree-level-order-traversal/

来源：力扣(LeetCode)

链接：https://leetcode-cn.com/problems/cong-shang-dao-xia-da-yin-er-cha-shu-ii-lcof

著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

题解

/**

* Definition for a binary tree node.

* public class TreeNode {

* int val;

* TreeNode left;

* TreeNode right;

* TreeNode(int x) { val = x; }

* }

*/

class Solution {

public List> levelOrder(TreeNode root) {

if (root == null) {

return Collections.emptyList();

}

Queue queue = new LinkedList();

queue.offer(root);

List> result = new ArrayList();

while (!queue.isEmpty()) {

List list = new ArrayList<>();

int length = queue.size();

for (int i=0; i< length; i++) {

TreeNode node = (TreeNode)queue.poll();

if (node != null) {

list.add(node.val);

queue.offer(node.left);

queue.offer(node.right);

}

}

if (!list.isEmpty()){

result.add(list);

}

}

return result;

}

}

小结

这是二叉树层次遍历的一个变种，区别在于每次poll的之前需要先记录下当前queue的size，即当前层的节点个数，然后按这个size去pull。

doc