按之字形顺序打印二叉树

按之字形顺序打印二叉树

http://www.nowcoder.com/questionTerminal/91b69814117f4e8097390d107d2efbe0

思路:其实就是树的广度遍历,不过每一层要进行一次反向即可。当然可以用queue来做,也可以通过计算树每个节点所在高度来选择是否反转此层。而我这里是通过两个栈来实现的,不需要计算高度,也不需要用标记来标记哪一层需要反转:
这个办法我写起来其实蛮麻烦的,相信有更简单的代码能实现这个思路。

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) {
        Stack<TreeNode> stack1 = new Stack<TreeNode>();//遍历奇数层
        Stack<TreeNode> stack2 = new Stack<TreeNode>();//遍历偶数层
        ArrayList<ArrayList<Integer>> results = new ArrayList<ArrayList<Integer>>();
        ArrayList<Integer> result = new ArrayList<Integer>();
        if(pRoot == null){
            return results;
        }
        stack1.push(pRoot);
        while(stack1.size()!=0 || stack2.size()!=0){
            if(stack1.size()!=0){//stack1不为空,就一直出栈,并把左右子压入stack2
                TreeNode node = stack1.peek();
                if(node.left!=null){
                    stack2.push(node.left);//先入左再入右,则根据stack会反向
                }
                if(node.right!=null){
                    stack2.push(node.right);
                }
                result.add(stack1.pop().val);//将出栈的节点记录到result中
                if(stack1.size()==0){//当栈出完,那么就将result记录到results中
                    //注意,因为result要重用,故这里定义一个currentResult来保留result中的值
                    ArrayList<Integer> currentResult = new ArrayList<Integer>();
                    for(int i=0;i<result.size();i++) {
                        currentResult.add(result.get(i));
                    }
                    results.add(currentResult);
                    while(result.size()>0){//将result值清空,供下一层重用
                        result.remove(0);
                    }
                }
            }else{//stack2.size()!=0
                while(true) {//用while是为了防止stack2栈没出完就又去出stack1栈了
                    TreeNode node = stack2.peek();
                    if(node.right!=null){
                        stack1.push(node.right);//先右后左,再次反向
                    }
                    if(node.left!=null){
                        stack1.push(node.left);
                    }
                    result.add(stack2.pop().val);
                    if(stack2.size()==0){//出完栈,则记录值,并且break这个while循环,换到stack1进行出栈
                        ArrayList<Integer> currentResult = new ArrayList<Integer>();
                        for(int i=0;i<result.size();i++) {
                            currentResult.add(result.get(i));
                        }
                        results.add(currentResult);
                        while(result.size()>0){
                            result.remove(0);
                        }
                        break;
                    }
                }
            }
        }
        return results;
    }

}
全部评论
第二层的for可以用while来做,比较简单
点赞 回复 分享
发布于 2020-04-09 20:08

相关推荐

最近又搬回宿舍了,在工位坐不住,写一写秋招起伏不断的心态变化,也算对自己心态的一些思考表演式学习从开始为实习准备的时候就特别焦虑,楼主一开始选择的是cpp后端,但是24届这个方向已经炸了,同时自己又因为本科非92且非科班,所以感到机会更加迷茫。在某天晚上用java写出hello&nbsp;world并失眠一整晚后选择老本行干嵌入式。理想是美好的,现实情况是每天忙但又没有实质性进展,总是在配环境,调工具,顺带还要推科研。而这时候才发现自己一直在表演式学习,徘徊在设想如何展开工作的循环里,导致没有实质性进展。现在看来当时如果把精力专注在动手写而不是两只手端着看教程,基本功或许不会那么差。实习的焦虑5月,楼主...
耶比:哲学上有一个问题,玛丽的房间:玛丽知道眼睛识别色彩的原理知道各种颜色,但是她生活在黑白的房间里,直到有一天玛丽的房门打开了她亲眼看到了颜色,才知道什么是色彩。我现在最大可能的减少对非工作事情的思考,如果有一件事困扰了我, 能解决的我就直接做(去哪里或者和谁吵架等等……),解决不了的我就不想了,每一天都是最年轻的一天,珍惜今天吧
投递比亚迪等公司10个岗位 > 秋招被确诊为…… 牛客创作赏金赛
点赞 评论 收藏
分享
点赞 评论 收藏
分享
评论
1
收藏
分享
牛客网
牛客企业服务