之打印二叉树算法

最新推荐文章于 2021-10-20 20:27:37 发布

nzch

最新推荐文章于 2021-10-20 20:27:37 发布

阅读量168

点赞数

分类专栏：刷题

本文链接：https://blog.csdn.net/qq_28764557/article/details/119545899

版权

刷题专栏收录该内容

18 篇文章 0 订阅

订阅专栏

http://www.manongjc.com/detail/10-bnbcadnmhcmqdya.html

地址：https://www.nowcoder.com/practice/91b69814117f4e8097390d107d2efbe0?tpId=117&&tqId=37722&rp=1&ru=/activity/oj&qru=/ta/job-code-high/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 {
    public ArrayList<ArrayList<Integer> > Print(TreeNode pRoot) {
        // 开始需要初始化 两个栈先进先出的 层数1 入栈根节点
        Stack<TreeNode> s1 = new Stack<TreeNode>();
        Stack<TreeNode> s2 = new Stack<TreeNode>();
        int layer = 1;
        s1.push(pRoot);
        ArrayList<ArrayList<Integer>> list = new ArrayList<ArrayList<Integer>>();
        // while循环 任意一个栈不是空就是没打印完毕
        while(!s1.isEmpty()||!s2.isEmpty()){
            // 在奇数层则打印的是先左节点后右节点
            if(layer%2!=0){
               ArrayList<Integer> temp = new ArrayList<>();
               while(!s1.empty()){
                   TreeNode node = s1.pop();
                   if(node!=null){
                   temp.add(node.val);
                   s2.push(node.left);
                   s2.push(node.right);
                   }
               }
               if(!temp.isEmpty()){
                   list.add(temp);
                   layer++;
               }
            }else{
                // 偶数层打印的是先右节点后左节点
                ArrayList<Integer> temp = new ArrayList<>();
                 while(!s2.empty()){
                   TreeNode node = s2.pop();
                   if(node!=null){
                   temp.add(node.val);
                   s1.push(node.right);
                   s1.push(node.left);
                   }
               }
               if(!temp.isEmpty()){
                   list.add(temp);
                   layer++;
               }
            }
        }
        return list;
    }

}