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=new LinkedList();
②弹出队列元素：queue.poll();
③两层while循环