给定一个二叉树,返回其按层次遍历的节点值。 (即逐层地,从左到右访问所有节点)。
例如:
给定二叉树: [3,9,20,null,null,15,7]
,
3 / \ 9 20 / \ 15 7
返回其层次遍历结果:
[ [3], [9,20], [15,7] ]
package LeetCode;
import java.util.ArrayList;
import java.util.LinkedList;
import java.util.List;
import java.util.Queue;
public class LevelOrder {
public List<List<Integer>> levelOrder(TreeNode root) {
if (root == null) return null;
List<List<Integer>> res = new ArrayList<>();
Queue<TreeNode> queue = new LinkedList<>();
TreeNode r = root;
queue.add(r);
while (!queue.isEmpty()) {
int count = queue.size();
List<Integer> rs = new ArrayList<>();
while (count > 0) {
TreeNode n = queue.poll();
rs.add(n.val);
if (n.left != null) queue.add(n.left);
if (n.right != null) queue.add(n.right);
count--;
}
res.add(rs);
}
return res;
}
}