Binary Tree Level Order Traversal ---LeetCode

最新推荐文章于 2020-12-30 23:47:17 发布

Kexin_Li

最新推荐文章于 2020-12-30 23:47:17 发布

阅读量326

点赞数

分类专栏： Leetcode 文章标签： bfs 遍历二叉树 leetcode java

本文链接：https://blog.csdn.net/nichole_xx/article/details/53370763

版权

Leetcode 专栏收录该内容

50 篇文章 0 订阅

订阅专栏

该博客主要探讨了LeetCode上的二叉树层序遍历问题，利用广度优先搜索(BFS)策略解决。通过维护一个队列，从根节点开始，根据队列大小控制每层节点的遍历，实现层次遍历。

摘要由CSDN通过智能技术生成

https://leetcode.com/problems/binary-tree-level-order-traversal/

解题思路：

这道题是二叉树的层序遍历，即 BFS 的应用。

这里我们维护一个队列，先将根节点入列。在队列不为空的循环体中，依次加入每一层的左右节点，用队列的 size 来控制每一层的节点是否遍历结束。

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
public class Solution {
    public List<List<Integer>> levelOrder(TreeNode root) {
        List<List<Integer>> result = new ArrayList<>();
        Queue<TreeNode> queue = new LinkedList<>();

        if (root == null) return result;
        queue.add(root);
        while (!queue.isEmpty()) {
            List<Integer> list = new ArrayList<>();
            int size = queue.size();
            for (int i=0; i<size; i++) {
                root = queue.poll();
                list.add(root.val);
                if (root.left != null)
                    queue.add(root.left);
                if (root.right != null)
                    queue.add(root.right);
            }
            result.add(list);
        }
        return result;
    }
}