剑指Offer59:按之字形顺序打印二叉树(Java)

参考“牛客408964号”牛客解答:https://www.nowcoder.com/questionTerminal/91b69814117f4e8097390d107d2efbe0?f=discussion

思路分析:

利用栈先进后出的特性
偶数层 往栈以先右后左的顺序压入,当奇数层从栈中取值时得到的顺序时由左向右
奇数层 往栈以先左后右的顺序压入,当偶数层从栈中取值时得到的顺序时由右向左

题目描述

请实现一个函数按照之字形打印二叉树,即第一行按照从左到右的顺序打印,第二层按照从右至左的顺序打印,第三行按照从左到右的顺序打印,其他行以此类推。

Java代码:

import java.util.ArrayList;
import java.util.Stack;
public class Solution {
    public ArrayList > Print(TreeNode pRoot) {
    Stack s1=new Stack<>();
    Stack s2=new Stack<>();
    ArrayList> res=new ArrayList>();
    s1.push(pRoot);
    int layer=1;
        while(!s1.empty()||!s2.empty()){
            if(layer%2!=0){//奇数层,从s1取值遍历,孩子节点按照先左后右的顺序压入s2
                ArrayList tmp=new ArrayList();
                while(!s1.empty()){//把一个栈遍历完了,也就是等同于遍历了一层二叉树
                    TreeNode node=s1.pop();
                    if(node!=null){
                        tmp.add(node.val);//需要判断node的空与否,否则会报空指针
                        s2.push(node.left);
                        s2.push(node.right);
                    }
                }
                if(!tmp.isEmpty()){//注意判空,当是时是tmp是空的,会被添加进最后结果集。
                    res.add(tmp);
                    layer++;
                }
            }else{//偶数层,从s2取值遍历,孩子节点按照先右后左的顺序压入s1
                ArrayList tmp=new ArrayList();
                while(!s2.empty()){
                    TreeNode node=s2.pop();
                    if(node!=null){
                        tmp.add(node.val);
                        s1.push(node.right);
                        s1.push(node.left);
                    }
                }
                if(!tmp.isEmpty()){
                    res.add(tmp);
                    layer++;
                }
            }
        }
        return res;
    }
}

你可能感兴趣的:(剑指Offer59:按之字形顺序打印二叉树(Java))