【Leetcode】102. Binary Tree Level Order Traversal

167 阅读1分钟

题目地址: leetcode.com/problems/bi…

经典的树的广度优先遍历。用一个队列即可。基本思想是,每当从队列中poll一个数出来的时候,就将它的子节点offer进去,这样每一层遍历完后,队列里已经从左向右存储了当前这一层的所有子节点,于是就可以继续poll出来访问。需要注意的一个地方是,要提前存储队列的size,也就是当前这一层的节点数目,因为队列的size在poll的时候是动态变化的。代码如下:

import java.util.ArrayDeque;
import java.util.ArrayList;
import java.util.List;
import java.util.Queue;

public class Solution {
    public List<List<Integer>> levelOrder(TreeNode root) {
        List<List<Integer>> rows = new ArrayList<>();
        if (root == null) {
            return rows;
        }
        
        Queue<TreeNode> queue = new ArrayDeque<>();
        queue.offer(root);
        while (!queue.isEmpty()) {
            List<Integer> row = new ArrayList<>();
            // 提前存储当前queue的size,也就是当前这一层的节点个数
            int len = queue.size();
            for (int i = 0; i < len; i++) {
                TreeNode x = queue.poll();
                row.add(x.val);
                if (x.left != null) {
                    queue.offer(x.left);
                }
                if (x.right != null) {
                    queue.offer(x.right);
                }
            }
            rows.add(row);
        }
        return rows;
    }
}

class TreeNode {
    int val;
    TreeNode left;
    TreeNode right;
    TreeNode(int x) {
        val = x;
    }
}

  1. 相关链接:102. Binary Tree Level Order Traversal · leetcode
  2. (9条消息) 【Leetcode】102. Binary Tree Level Order Traversal_edWard的博客-CSDN博客