按之字形顺序打印二叉树

题目描述

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

 

设计思路

用两个栈分别村粗单数层和双数层的输出即可，并且发现之字形的输入只要改变栈push时候的进栈顺序就可以轻易实现。

代码

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) {

        int layer = 1;

        Stack<TreeNode> s1 = new Stack<TreeNode>();

        s1.push(pRoot);

        Stack<TreeNode> s2 = new Stack<TreeNode>();

        ArrayList<ArrayList<Integer>> list = new ArrayList<ArrayList<Integer>>();

        while(!s1.isEmpty() || !s2.isEmpty()){

            if(layer%2!=0){

                ArrayList<Integer> temp = new ArrayList<Integer>();

                while(!s1.isEmpty()){

                    TreeNode node = s1.pop();

                    if(node!=null){

                        temp.add(node.val);

                        System.out.print(node.val+" ");

                        s2.push(node.left);

                        s2.push(node.right);

                    }

                }

                if(!temp.isEmpty()){

                    list.add(temp);

                    layer++;

                    System.out.println();

                }

            }else{

                ArrayList<Integer> temp = new ArrayList<Integer>();

                while(!s2.isEmpty()){

                    TreeNode node = s2.pop();

                    if(node!=null){

                        temp.add(node.val);

                        System.out.print(node.val+" ");

                        s1.push(node.right);

                        s1.push(node.left);

                    }

                }

                if(!temp.isEmpty()){

                    list.add(temp);

                    layer++;

                    System.out.println();

                }

            }

          

            

            

        }

        

    }

}