按之字形顺序打印二叉树

思路:使用两个栈,从stackOut弹出结点并添加到当前层,将该结点的孩子结点按照下一轮打印顺序压入stackIn。

    public ArrayList<ArrayList<Integer> > Print(TreeNode pRoot) {
        if(pRoot==null) return new ArrayList<>();
        ArrayList<ArrayList<Integer>> answer=new ArrayList<>();
        Stack<TreeNode> stack1=new Stack<>();
        Stack<TreeNode> stack2=new Stack<>();
        //两个栈与打印方向
        Stack<TreeNode> stackOut=null;
        Stack<TreeNode> stackIn=null;
        boolean dirctionFlag=false;//false代表下一轮从右向左打印
        //tmp val
        TreeNode node=null;

        stack1.push(pRoot);
        while(!(stack1.empty()&&stack2.empty())){
            ArrayList<Integer> currentFloor=new ArrayList<>();
            stackOut=stack1.empty()?stack2:stack1;
            stackIn=stack1.empty()?stack1:stack2;
            //一轮打印
            while(!stackOut.empty()){
                node=stackOut.pop();
                currentFloor.add(node.val);
                if(dirctionFlag) {
                    if(node.right!=null) stackIn.push(node.right);
                    if(node.left!=null) stackIn.push(node.left);
                }
                else{
                    if(node.left!=null) stackIn.push(node.left);
                    if(node.right!=null) stackIn.push(node.right);
                }
            }
            answer.add(currentFloor);
            dirctionFlag=!dirctionFlag;
        }
        return answer;
    }

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值