树的bfs, 注意root可能是null
java
class TreeNode {
int val;
TreeNode left;
TreeNode right;
TreeNode() {}
TreeNode(int val) { this.val = val; }
TreeNode(int val, TreeNode left, TreeNode right) {
this.val = val;
this.left = left;
this.right = right;
}
}
// @lc code=start
/**
* Definition for a binary tree node.
*/
class Solution {
/**
* 树的bfs, 注意root可能是null
*/
public List<List<Integer>> levelOrder(TreeNode root) {
Deque<TreeNode> que = new ArrayDeque<>();
if (root != null) que.addLast(root);
List<List<Integer>> ans = new ArrayList<>();
while (!que.isEmpty()) {
List<Integer> arr = new ArrayList<>();
int cnt = que.size();
for (int i = 0; i < cnt; i++) {
TreeNode tmp = que.removeFirst();
if (tmp.left != null) que.addLast(tmp.left);
if (tmp.right != null) que.addLast(tmp.right);
arr.add(tmp.val);
}
ans.add(arr);
}
return ans;
}
}