LeetCode102: Binary Tree Level Order Traversal

最新推荐文章于 2024-07-11 23:32:17 发布

原创最新推荐文章于 2024-07-11 23:32:17 发布 · 391 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#bfs #leetcode

Leetcode 专栏收录该内容

9 篇文章

订阅专栏

该博客主要介绍了LeetCode第102题的解决方案，通过广度优先搜索（BFS）来实现二叉树的层级遍历。在遍历过程中，为确保每层节点完整输出，采用先获取队列当前长度，然后用for循环按层遍历的方式，避免不断添加节点到队列的操作。博主提供了AC代码。

题目要求：https://leetcode.com/problems/binary-tree-level-order-traversal/

1）此题本质上是一个BFS问题，不难解决

2）在对queue遍历时需要注意

题目要求将每一层节点作为一个list，那么使用while(!q.isEmpty())的方式来遍历Queue就不太合适，因为需要不停地将节点加入队列。
这里在访问每一层时，先获得此时queue里的长度，用for循环遍历。
使用Iterator()同样可行

AC代码：

/**
 * 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) {
        Queue<TreeNode> q = new LinkedList<TreeNode>();
        List<List<Integer>> resList = new LinkedList<List<Integer>>();
        if(root == null) return resList;
        q.add(root);
        
        while(!q.isEmpty()){
            List<Integer> levelList = new LinkedList<Integer>();
            int qSize = q.size();
            for(int i=0; i<qSize; i++){
                TreeNode tmp = q.poll();
                levelList.add(tmp.val);
                if(tmp.left != null){
                    q.add(tmp.left);
                }
                if(tmp.right != null){
                    q.add(tmp.right);
                }
            }
            resList.add(levelList);
        }
        return resList;
    }
}