按之字形顺序打印二叉树

按之字形顺序打印二叉树

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

相关推荐

不愿透露姓名的神秘牛友
10-05 10:13
已编辑
HHHHaos:让这些老登来现在秋招一下,简历都过不去
点赞 评论 收藏
分享
1 收藏 评论
分享
牛客网
牛客企业服务