LeetCode 102:二叉树的层序遍历
题目
给你二叉树的根节点 root ,返回其节点值的 层序遍历 。 (即逐层地,从左到右访问所有节点)。
示例 1:
输入:root = [3,9,20,null,null,15,7]
输出:[[3],[9,20],[15,7]]
示例 2:
输入:root = [1]
输出:[[1]]
示例 3:
输入:root = []
输出:[]
思路
利用队列性质
代码
class Solution {
public List<List<Integer>> levelOrder(TreeNode root) {
if(root == null)//考虑root为空
return new ArrayList<>();
List<List<Integer>> res=new ArrayList<>();
Queue<TreeNode> queue=new LinkedList<TreeNode>();
queue.add(root);
while(!queue.isEmpty())
{
List<Integer> p=new ArrayList<>();
int count=queue.size();
while(count>0)//利用count实现层序遍历
{
TreeNode t=queue.poll();
p.add(t.val);
if(t.left!=null)
queue.add(t.left);
if(t.right!=null)
queue.add(t.right);
count--;
}
res.add(p);
}
return res;
}
}
反思
①创建队列语句:Queue
②弹出队列元素:queue.poll();
③两层while循环