剑指offer 44. 分行从上往下打印二叉树-java

AcWing 44. 分行从上往下打印二叉树

原题链接

从上到下按层打印二叉树，同一层的结点按从左到右的顺序打印，每一层打印到一行。

数据范围
树中节点的数量 [0,1000]。

代码案例：

题解

通过在每一行中添加Null 来确定这是否是一行的界限

写法1

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public List<List<Integer>> printFromTopToBottom(TreeNode root) {
        List<List<Integer>> res = new ArrayList<>();
        if(root == null ) return res;
        Queue<TreeNode> q =  new LinkedList<>();
        q.offer(root);
           //采用加入null，来标志本层元素已经遍历完
        q.add(null);
        List<Integer> level = new ArrayList<>();//储存每行
        while(!q.isEmpty()){
            TreeNode t = q.poll();
            if(t == null){//遍历完一整行
              if(level.isEmpty()) break ;//下一行什么都没有  就是level为空的情况
              //并且如果level集合大小为0，说明整个树已经遍历完了，直接退出循环即可
               
                //否则说明当前层的所有节点已经遍历完，添加当前层所有元素到返回集合中
                //这里创建一个新的集合的原因：就是res这个集合的为一个集合元素 还是一直对 level集合保持引用关系
                //如果level改变，res集合里面的每一个都会对应改变。
               List<Integer> tmp = new ArrayList<>(level);
              res.add(tmp);
              level.clear(); //清除集合所有元素，为了下一层元素进入
              q.offer(null);//结尾加上Null
              continue;//直接进入下一次循环，因为当前元素是null
            }
            level.add(t.val);
            if(t.left != null) q.offer(t.left);
            if(t.right != null) q.offer(t.right);
            
            
        }
        return res ;
    }
}

写法2 通过q.size(）遍历次数

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public List<List<Integer>> printFromTopToBottom(TreeNode root) {
        List<List<Integer>> res = new ArrayList<>();
        Queue<TreeNode> q = new LinkedList<>();
        if(root != null) q.add(root);
        while(!q.isEmpty()) {
            List<Integer> level = new ArrayList<>();
            // 这里采用len表示这一层的所有节点个数。也可以在每一层最后面加一个null，来标志这一层结束
            int len = q.size();
            while(len -- > 0) {
                TreeNode t = q.poll();
                level.add(t.val);
                if(t.left != null) q.add(t.left);
                if(t.right != null) q.add(t.right); 
            }
            res.add(level);
        }
        return res;

 
    }
}