剑指Offer-二叉树-（11）

知识点/数据结构：二叉树

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

注意点

左右轮流往数据容器中塞数据的过程很妙！

import java.util.ArrayList;
import java.util.Stack;


/*
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) {
        //先定义一个存放返回值的容器
        ArrayList<ArrayList<Integer>> list=new ArrayList<>();
        if(pRoot==null)   return list;
            //定义两个栈，轮流打印每一层，根据栈的特点来添加到打印某一次的链表中
           Stack<TreeNode> s1=new Stack<TreeNode>();
           s1.add(pRoot);
           Stack<TreeNode> s2=new Stack<TreeNode>();
        
           while(!s1.isEmpty()||!s2.isEmpty()){
               if(!s1.isEmpty()){
                   //下面这个容器就是每次  左或者右打印一次的容器。
                   ArrayList<Integer> listt=new ArrayList<Integer>();
                   while(!s1.isEmpty()){
                       //这个时候s1也就空了
                       TreeNode p=s1.pop();
                       //打印树节点就是把它的值输出
                       listt.add(p.val);
                       //如果有下一层，添加到栈2中
                       if(p.left!=null)
                           s2.add(p.left);
                       if(p.right!=null)
                           s2.add(p.right);
                   }
                   //从左到右的一次节点值，存放到总的容器中
                   list.add(listt);
               }else {
                   ArrayList<Integer> listtt=new ArrayList<Integer>();
                   while(!s2.isEmpty()){
                      TreeNode p=s2.pop();
                      listtt.add(p.val);
                   //这个时候出栈一次，然后while循环，出栈直到空了。
                   //下面的操作就是这个节点的左右子节点再次添加到s1中，注意添加顺序，和上面正好相反
                      if(p.right!=null)  s1.add(p.right);
                      if(p.left!=null)   s1.add(p.left);              
                    }
                    list.add(listtt);
                }
           }
           return list;
    }

}