剑指Offer——把二叉树打印成多行

最新推荐文章于 2021-10-13 16:29:09 发布

原创最新推荐文章于 2021-10-13 16:29:09 发布 · 200 阅读

0 ·

CC 4.0 BY-SA版权

剑指Offer 专栏收录该内容

65 篇文章

订阅专栏

本文介绍了一种从上到下按层打印二叉树的方法，确保同一层节点从左至右输出，每层输出一行。通过使用队列跟踪当前层和下一层的节点数，实现了层次遍历。

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

思路：同从上向下打印二叉树思路，但是需要添加两个变量记录当前层未打印节点和下一层需要打印的节点

代码：

import java.util.ArrayList;
import java.util.Queue;
import java.util.LinkedList;
/*
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>> Alist=new  ArrayList<ArrayList<Integer>>();
        if(pRoot==null){
            return Alist;
        }
        Queue<TreeNode> queue =new LinkedList();
        queue.add(pRoot);
        int toBePrint=queue.size();
        int nextLevel=0;
        while(!queue.isEmpty()){
            ArrayList<Integer> aList=new ArrayList<Integer>();
            while(toBePrint!=0){
                if(queue.peek().left!=null){
                    queue.add(queue.peek().left);
                    nextLevel++;
                }
                if(queue.peek().right!=null){
                    queue.add(queue.peek().right);
                    nextLevel++;
                }
                aList.add(queue.poll().val);
                toBePrint--;
            }
            Alist.add(aList);
            toBePrint=nextLevel;
            nextLevel=0;
        }
        return Alist;
    }
}