面试题60：把二叉树打印成多行

最新推荐文章于 2019-05-09 13:41:53 发布

原创最新推荐文章于 2019-05-09 13:41:53 发布 · 241 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#二叉树

剑指offer 专栏收录该内容

68 篇文章

订阅专栏

本文介绍了一种使用Java实现的二叉树层次遍历算法。通过定义一个节点类和解决方案类，实现了按层次顺序打印二叉树节点值的功能。算法采用队列的方式逐层处理节点，最终将每层的节点值收集到一个二维数组中。

import java.util.ArrayList;
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>> result =new  ArrayList<ArrayList<Integer>>();
        if(pRoot==null)
            return result;
        LinkedList<TreeNode> layer = new LinkedList<TreeNode> ();       
        ArrayList<Integer> layerList=new ArrayList<Integer> ();
        layer.add(pRoot);
        
        int start=0,end=1;
        while(!layer.isEmpty()){
            TreeNode cur=layer.remove();
            layerList.add(cur.val);
            
            if(cur.left!=null){
                start++;
                layer.add(cur.left);
            }
            if(cur.right!=null){
                start++;
                layer.add(cur.right);
            }
            
            end--;
            if(end==0){
                
                end=start;
                start=0;
                result.add(layerList);
                layerList=new ArrayList<Integer> ();
            }
        }
        return result;
    }
    
}