讨论/《零起步学算法》 - 例题:二叉树的层序遍历/
《零起步学算法》 - 例题:二叉树的层序遍历
共 1 个回复

思考:queue的add和offer都可以添加元素,为什么推荐用offer?原因:当容器满的时候,add会抛出异常,offer只会返回false

import java.util.LinkedList;
 import java.util.ArrayList;
 import java.util.Queue;
 import java.util.List;
class Solution {
    public List<List<Integer>> levelOrder(TreeNode root) {
      //广度优先遍历 => 层级遍历
      //创建链表
      List<List<Integer>> res = new LinkedList<>();
      //判空
      if(root == null) {
        return res;
      }
      //队列
      Queue<TreeNode> queue = new LinkedList<>();
      //根结点放入队列
      queue.add(root);
      while(!queue.isEmpty()) {
        //将元素放入这个list中
        List<Integer> curLevel = new ArrayList<>();
        //长度
        int curSize = queue.size();
        //循环遍历
        for(int i = 0; i < curSize; i++) {
          //出队列
          TreeNode head = queue.poll();
          //放入list
          curLevel.add(head.val);
          if(head.left != null) {
            //左节点,防止
            queue.offer(head.left);
          }
          if(head.right != null) {
            //右节点
            queue.offer(head.right);
          }
        }
        res.add(curLevel);
      }
      return res;
    }
}
2