LeetCode102. 二叉树的层序遍历

题目描述

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

涉及tag

二叉树

算法思路

本题借助数据结构队列，借助队列的先进先出特性，将根节点、根节点的下一层所有节点（从左往右）送入队列，最终得到层序遍历之后的新树。
首先将根节点入queue，变量n存放每一层的节点数，node变量存放新树的每一颗子树的根节点，level存放每一层节点的val值，res = sum（level），返回res。

示例代码

class Solution {
    public List<List<Integer>> levelOrder(TreeNode root) {
       List<List<Integer>> res = new ArrayList<>();
       Queue<TreeNode> queue = new ArrayDeque<>();
       if (root != null) {
           queue.add(root);
       }
       while (!queue.isEmpty()) {
           //n表示每一层的节点个数
           int n = queue.size();
           //level存储节点val值
           List<Integer> level = new ArrayList<>();
           for (int i = 0; i < n; i++) {
               //node作为一个形参，相当于每个子树的根节点
               TreeNode node = queue.poll();
               level.add(node.val);
               if (node.left != null) {
                   //将遍历得到的左子树加入到新的queue中
                   queue.add(node.left);
               }
               if (node.right != null) {
                   queue.add(node.right);
               }
           }
           //每一层level集合后放入res
           res.add(level);
       }
       return res;
    }
}