102. 二叉树的层序遍历
给你一个二叉树,请你返回其按 层序遍历 得到的节点值。 (即逐层地,从左到右访问所有节点)。
示例:
二叉树:[3,9,20,null,null,15,7],
返回其层序遍历结果
思路:BFS(广度优先搜索),再考虑到输出格式,采用二维数组ans存储每一层的数组,同时为了保证将同一层的值写进level里,利用for循环,将确保每一层的全部输出,在 while 循环的每一轮中,都是将当前层的所有结点出队列,再将下一层的所有结点入队列,这样就实现了层序遍历。
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public List<List<Integer>> levelOrder(TreeNode root) {
List<List<Integer>> ans = new ArrayList<>();
Queue<TreeNode> queue = new ArrayDeque();
if(root!=null)
queue.add(root);
while(!queue.isEmpty())
{
int n= queue.size();
List<Integer> level= new ArrayList<>();
for(int i=0;i<n;i++)
{
TreeNode node = queue.poll();
level.add(node.val);
if(node.left!=null)
{
queue.add(node.left);
}
if(node.right!=null)
{
queue.add(node.right);
}
}
ans.add(level);
}
return ans;
}
// public void bfs(TreeNode root)
// {
// Queue<TreeNode> queue = new ArrayDeque<>();
// queue.add(root);
// while(!queue.isEmpty())
// {
// TreeNode node =queue.poll();
// if(node.left!=null)
// queue.add(node.left);
// if(queue.right!=null)
// queue.add(node.right);
// }
// }
}