LeetCode-102-二叉树的层序遍历 LeetCode-102-二叉树的层序遍历

二叉树的层序遍历

题目描述：给你一个二叉树，请你返回其按层序遍历得到的节点值。（即逐层地，从左到右访问所有节点）。
示例说明请见LeetCode官网。
来源：力扣（LeetCode）
链接：https://leetcode-cn.com/probl...
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解法一：利用队列遍历二叉树

队列的特点是先进先出，所以利用队列来遍历二叉树实现层序遍历，具体过程如下：

首先，如果root为空，直接返回空的List。

【LeetCode-102-二叉树的层序遍历】如果root不为空，声明一个队列nodes，将root节点加入到队列中，声明一个result，然后遍历nodes中的节点，遍历过程如下：

首先，用count记录当前队列中节点的数量，即为当前层节点的数量，vals记录当前节点的值；

从nodes中按顺序取出count个节点，并且将相应的节点值放入vals中，同时如果当前节点的左右子节点不为空，按从左到右的顺序放入nodes中。然后将vals加入到结果result中。

重复遍历nodes中的节点，直到nodes为空。

最后返回result即为层序遍历的结果。

import java.util.ArrayList; import java.util.LinkedList; import java.util.List; import java.util.Queue; public class LeetCode_102 { /** * 利用队列遍历二叉树：队列的特性是先进先出 * @param root * @return */ public static List> levelOrder(TreeNode root) { if (root == null) { return new ArrayList<>(); } List> result = new ArrayList<>(); // 暂存每一行的节点 Queue nodes = new LinkedList<>(); nodes.add(root); while (!nodes.isEmpty()) { // 每次遍历一行数据 List vals = new ArrayList<>(); int count = nodes.size(); while (count > 0) { TreeNode curNode = nodes.poll(); vals.add(curNode.val); // 将当前节点的左右子节点从左到右放入队列 if (curNode.left != null) { nodes.add(curNode.left); } if (curNode.right != null) { nodes.add(curNode.right); } count--; } result.add(vals); } return result; }public static void main(String[] args) { TreeNode root = new TreeNode(3); root.left = new TreeNode(9); root.right = new TreeNode(20); root.right.left = new TreeNode(15); root.right.right = new TreeNode(7); for (List integers : levelOrder(root)) { for (Integer integer : integers) { System.out.print(integer + " "); } System.out.println(); } } }