[LeetCode] 102. 二叉树的层序遍历

题外话(BFS 的使用场景总结:层序遍历、最短路径问题

用队列存储,然后每次取得队列长度,把子节点放进去,取出相应得节点

有想法不代表能写的出来

public List<List<Integer>> levelOrder(TreeNode root) {
    List<List<Integer>> res = new ArrayList<>();

    Queue<TreeNode> queue = new ArrayDeque<>();
    if (root != null) {
        queue.add(root);
    }
    while (!queue.isEmpty()) {
        int n = queue.size();
        List<Integer> level = new ArrayList<>();
        for (int i = 0; i < n; i++) { 
            TreeNode node = queue.poll();
            level.add(node.val);
            if (node.left != null) {
                queue.add(node.left);
            }
            if (node.right != null) {
                queue.add(node.right);
            }
        }
        res.add(level);
    }

    return res;
}

猜你喜欢

转载自www.cnblogs.com/doyi111/p/12903221.html