【编程题】把二叉树打印成多行（java实现）

最新推荐文章于 2024-03-24 21:42:51 发布

xclia

最新推荐文章于 2024-03-24 21:42:51 发布

阅读量198

点赞数

分类专栏：编程打卡

本文链接：https://blog.csdn.net/qq_40432108/article/details/98617705

版权

编程打卡专栏收录该内容

90 篇文章 0 订阅

订阅专栏

【编程题】把二叉树打印成多行（java实现）

题目来源

剑指offer
https://www.nowcoder.com/practice/445c44d982d04483b04a54f298796288?tpId=13&tqId=11213&rp=4&ru=/ta/coding-interviews&qru=/ta/coding-interviews/question-ranking

题目描述

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

解答

方法一

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>>printlist=new ArrayList<ArrayList<Integer>>();
         if(pRoot==null)
             return printlist;
        LinkedList<TreeNode>list=new LinkedList<>();
        list.add(pRoot);
        while(!list.isEmpty()){
            LinkedList<TreeNode>tem=new LinkedList<>();
            ArrayList<Integer>innerList=new ArrayList<>();
            while(!list.isEmpty()){
               pRoot=list.pollFirst();
               if(pRoot.left!=null)
                  tem.add(pRoot.left);
               if(pRoot.right!=null)
                  tem.add(pRoot.right);
              innerList.add(pRoot.val);
            }
           printlist.add(new ArrayList<>(innerList));
           list=tem;
        }
        return printlist;
    }
    
}

方法二

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>>printlist=new ArrayList<ArrayList<Integer>>();
         if(pRoot==null)
             return printlist;
        LinkedList<TreeNode>list=new LinkedList<>();
       ArrayList<Integer>innerList=new ArrayList<>();
        int start=0,end=1;
        list.add(pRoot);
        while(!list.isEmpty()){
            pRoot=list.pollFirst();
            innerList.add(pRoot.val);
            start++;
            if(pRoot.left!=null)
              list.add(pRoot.left);
            if(pRoot.right!=null)
              list.add(pRoot.right);
            if(start==end){
                end=list.size();
                start=0;
                printlist.add(new ArrayList<>(innerList));
                innerList=new ArrayList<>();
            }
        }
        return printlist;
    }
}