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

Posted 2021-09-03 雄狮虎豹

二叉树的层序遍历

题目描述：给你一个二叉树，请你返回其按 层序遍历 得到的节点值。 （即逐层地，从左到右访问所有节点）。

示例说明请见LeetCode官网。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/probl...
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

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

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

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

• 如果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<List<Integer>> levelOrder(TreeNode root) {
        if (root == null) {
            return new ArrayList<>();
        }
        List<List<Integer>> result = new ArrayList<>();
        // 暂存每一行的节点
        Queue<TreeNode> nodes = new LinkedList<>();
        nodes.add(root);
        while (!nodes.isEmpty()) {
            // 每次遍历一行数据
            List<Integer> 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<Integer> integers : levelOrder(root)) {
            for (Integer integer : integers) {
                System.out.print(integer + " ");
            }
            System.out.println();
        }
    }
}

【每日寄语】 虚心使人进步，骄傲使人落后。