二叉树的层序遍历-Java

目录

一、题目描述

二、运行结果

三、解题思路

四、代码

---

一、题目描述

本文代码为力扣102题解题代码，也是通用的二叉树层次遍历代码：

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

示例 1：

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

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

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

二、运行结果

三、解题思路

层序遍历是二叉树中比较典型的一个算法，需要用到一个辅助队列，思路是对二叉树的节点进行逐层处理。首先将头结点入队，头结点所在的层必定只有一个节点，每次求队列的长度得到的就是当前层二叉树节点的个数，然后逐个处理当前层结点（保存结点值，将其左右孩子结点入队（如果非空）），每处理完一个队头节点之后，就将其出队，这样就保证了处理完当前层后，队列中的所有结点都是下一层的结点，并且是按照从左到右的顺序排列，重复上述处理过程，直至队列为空。

四、代码

class Solution {
    public List<List<Integer>> levelOrder(TreeNode root) {
        List<List<Integer>> ans = new ArrayList<>(); //初始化二维列表
        if(root == null)
            return ans;
        Deque<TreeNode> d = new ArrayDeque<TreeNode>(); //辅助队列，存放每层节点
        d.addLast(root);
        while(!d.isEmpty()){
            int size = d.size();  //当前层的节点数
            List<Integer> list = new ArrayList<Integer>();
            for(int i=0; i<size; ++i){
                TreeNode node = d.pollFirst(); //取出队列的头节点
                list.add(node.val);
                if(node.left != null)
                    d.addLast(node.left);
                if(node.right != null)
                    d.addLast(node.right);
            }
            ans.add(list);  //一层的节点
        }
        return ans;
    }
}