目录
一、题目描述
本文代码为力扣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;
}
}