二叉树----6.二叉树的层序遍历

 题目链接

/**

            二叉树这个数据结构本身无法做到按层级遍历

            引入队列,将节点保存到队列中, 弹出节点时将该节点的子节点入队

            层级区分: 提前记录队列中元素个数

*/

/**
 * 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) {
        /**
            二叉树这个数据结构本身无法做到按层级遍历
            引入队列,将节点保存到队列中, 弹出节点时将该节点的子节点入队
            层级区分: 提前记录队列中元素个数
         */

         if(root == null) {
            return new ArrayList<>(); 
         }

        //按层级保存结果
        List<List<Integer>> ans = new ArrayList<>();
        //队列
        Queue<TreeNode> queue = new ArrayDeque<>();
        queue.add(root);
        
        while(!queue.isEmpty()) {
            //记录每层节点数
            int size = queue.size();
            //保存该层元素
            List<Integer> list = new ArrayList<>();

            //处理当前层  弹出该层节点，保存到list中 并将该节点的子节点入队
            while(size > 0) {
                //弹出节点 保存到数组中
                TreeNode node = queue.poll();
                list.add(node.val);

                //将弹出节点的子节点添加到队列中(通过提前记录size,区分层级)
                if(node.left != null) queue.add(node.left);
                if(node.right != null) queue.add(node.right);
                size--;
            }

            //将该层元素添加到ans中
            ans.add(list);
        }

        return ans;
    }
}