二叉树的层序遍历

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/binary-tree-level-order-traversal

给你二叉树的根节点 root ，返回其节点值的 层序遍历 。 （即逐层地，从左到右访问所有节点）。

点击查看代码

//递归
/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    public List<List<Integer>> levelOrder(TreeNode root) {
        //用来存放最终的结果
        List<List<Integer>> nodess = new ArrayList<>();
        if(root == null){
            return nodess;
        }
        //递归 存放结果的集合，当前的层级，当前的节点
        a(nodess,1,root);
        //返回结果
        return nodess;
    }
     //递归                    存放结果的集合，当前的层级，当前的节点
    public void a(List<List<Integer>> nodess,int index,TreeNode node){
        //创建一个空集合，当前的集合大小小于索引时
         List<Integer> nodes = new ArrayList<>();
         if(nodess.size() < index){
             nodess.add(nodes);
         }
        //将当前的节点值加入当前层级的集合
         nodess.get(index - 1).add(node.val);
         //判断当前左字节点是否不为空，递归
         if(node.left != null){
             a(nodess,index + 1,node.left);
         }
          //判断当前右字节点是否不为空，递归
         if(node.right != null){
             a(nodess,index + 1,node.right);
         }
            
    }
}

//迭代
class Solution {
    public List<List<Integer>> levelOrder(TreeNode root) {
        //用来存放最终的结果
        List<List<Integer>> nodess = new ArrayList<>();
        if(root == null){
            return nodess;
        }
       //创建存放各个节点的队列
       LinkedList<TreeNode> queue = new LinkedList<>();
       //将根节点加入队列，不断遍历队列，迭代
       queue.add(root);
       while(queue.size() > 0){
           //获取当前队列的长度，相当于当前层级的节点数
           int size = queue.size();
           //创建存放当前队列中节点值的集合
           ArrayList<Integer> nodes = new ArrayList<>();
           //如果节点的左右字节点不为空也放入队列中
           for(int i = 0;i<size;i++){
               TreeNode tmp = queue.remove();
               nodes.add(tmp.val);
               //如果当前左字节点不为空，将其加入队列
               if(tmp.left != null){
               queue.add(tmp.left);
              }
              //右子节点不为空，也加入队列
              if(tmp.right != null){
                  queue.add(tmp.right);
              }
           }
           //当前循环结束，将当前层级节点值加入结果集合中国
           nodess.add(nodes);
       }
        //返回结果
        return nodess;
    }
     
}

示例 1：

输入：root = [3,9,20,null,null,15,7]
输出：[[3],[9,20],[15,7]]
示例 2：

输入：root = [1]
输出：[[1]]
示例 3：

输入：root = []
输出：[]