剑指 Offer 32 - II. 从上到下打印二叉树 II

痛定思痛。 2022-12-15 12:46 112阅读 0赞

# 题目 #

从上到下按层打印二叉树，同一层的节点按从左到右的顺序打印，每一层打印到一行。

例如:  
给定二叉树: \[3,9,20,null,null,15,7\],  
![在这里插入图片描述][20201016174555836.png_pic_center]  
返回其层次遍历结果：  
![在这里插入图片描述][20201016174639133.png_pic_center]  
提示：  
节点总数 <= 1000

来源：力扣（LeetCode）  
链接：[https://leetcode-cn.com/problems/cong-shang-dao-xia-da-yin-er-cha-shu-ii-lcof][https_leetcode-cn.com_problems_cong-shang-dao-xia-da-yin-er-cha-shu-ii-lcof]

# 解题思路 #

本质还是二叉树的层序遍历,使用队列解决

# 代码 #

/** * 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>> resultList = new ArrayList<>();
    
            // 校验
            if (root == null) { 
                return resultList;
            }
    
            // 创建一个队列
            LinkedList<TreeNode> queue = new LinkedList<>();
            queue.add(root);
    
            // 判断队列是否为空
            while(!queue.isEmpty()) { 
                int size = queue.size();
                List<Integer> tempList = new ArrayList<>();
                // 循环
                while (size > 0) { 
                    // 出队
                    TreeNode node = queue.poll();
                    tempList.add(node.val);
                    // 添加左子树
                    if (node.left != null) { 
                        queue.add(node.left);
                    }
                    // 添加右子树
                    if (node.right != null) { 
                        queue.add(node.right);
                    }
                    size--;
                }
                // 一层遍历完成
                resultList.add(tempList);
            }
    
            return resultList;
        }
    }

[20201016174555836.png_pic_center]: /images/20221123/95d2165e7f334519b219c5275db3a02b.png
[20201016174639133.png_pic_center]: /images/20221123/13a03e53ec3c42fc974120faa26665c1.png
[https_leetcode-cn.com_problems_cong-shang-dao-xia-da-yin-er-cha-shu-ii-lcof]: https://leetcode-cn.com/problems/cong-shang-dao-xia-da-yin-er-cha-shu-ii-lcof