62.把二叉树打印成多行

Posted 2021-12-15 yjxyy

题目描述：

??从上到下按层打印二叉树，同一层结点从左至右输出。每一层输出一行。

思路分析：

??每一层都从左到右输出，我们可以用队列保存层序遍历的节点，这道题的难点在于，如何判断每一层节点的数目，以达到按层输出的目标，我们可以定义两个变量，nownum,和nextnum,分别记录当前层的节点数，和下一层的节点数。当nownum减为0的时候表示当前层遍历完。

代码：

import java.util.*;
/*
public class TreeNode 
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;

    public TreeNode(int val) 
        this.val = val;

    


*/
public class Solution 
    ArrayList<ArrayList<Integer> > Print(TreeNode pRoot) 
        ArrayList<ArrayList<Integer>>res=new ArrayList<>();
        ArrayList<Integer>list=new ArrayList<>();
        if(pRoot==null)
            return res;
        Queue<TreeNode>q=new LinkedList<>();
        int nownum=1;
        int nextnum=0;
        q.offer(pRoot);
        while(!q.isEmpty())
            TreeNode pNode=q.poll();
            nownum--;
            list.add(pNode.val);
            if(pNode.left!=null)
                q.offer(pNode.left);
                nextnum++;
            
            if(pNode.right!=null)
                q.offer(pNode.right);
                nextnum++;
            
            if(nownum==0)
                res.add(list);
                list=new ArrayList<>();
                nownum=nextnum;     //一层遍历完后更新nownum的值，将nextnum清空。
                nextnum=0;
            
        
        return res;