java刷题--102二叉树的层序遍历

Posted 2021-06-30 Anrys

java刷题--102二叉树的层序遍历

• 题目
• 代码
• 结果

题目

代码

class Solution {
    public List<List<Integer>> levelOrder(TreeNode root) {
        List<List<Integer>> res = new ArrayList<List<Integer>>();
        LinkedList<TreeNode> queue = new LinkedList<TreeNode>();
        if (root == null) return res;
        queue.add(root); //从上到下 先加入根节点 
        while (!queue.isEmpty()) {
            //每层新建一个level
            List<Integer> level = new ArrayList<Integer>();
            int size = queue.size();
            for (int i = 0; i < size; i++) {
                TreeNode node = queue.removeFirst();
                level.add(node.val);
                if (node.left != null) queue.add(node.left);
                if (node.right != null) queue.add(node.right);
            }res.add(level);
        }return res;
    }
}

结果