剑指OFFER-按之字形顺序打印二叉树(Java)

题目描述

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

核心代码实现

/*
public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;

    public TreeNode(int val) {
        this.val = val;

    }

}
*/
import java.util.*;

public class Solution {
    private int depth = 0;    //depth全局变量，用于计算二叉树的深度
    public ArrayList<ArrayList<Integer> > Print(TreeNode pRoot) {
        ArrayList<ArrayList<Integer> > pResult = new ArrayList<ArrayList<Integer> >();
        ArrayList<TreeNode> queue = new ArrayList<TreeNode>();    //借助队列进行层次遍历
        Stack<Integer> stack = new Stack<Integer>();    //偶数行借助栈从右到左打印
        if(pRoot == null){
            return pResult;    //空树
        }
        queue.add(pRoot);    //二叉树非空时，将根节点入队
        depth++;
        
        while(!queue.isEmpty()){    //判断非空：ArrayList用.isEmpty()，Stack用.empty();
            int size = queue.size();     //记录初始队列的长度
            pResult.add(new ArrayList<Integer>());    //新建行
            for(int i=0; i<size; i++){
                TreeNode node = queue.get(0);
                queue.remove(0);
                if(depth%2 == 1){    //奇数行，直接加入pResult
                    pResult.get(depth-1).add(node.val);
                }
                if(depth%2 == 0){    //偶数行，需要借助栈，从左往右依次压入栈
                    stack.push(node.val);
                }
                if(node.left != null){
                    queue.add(node.left);
                }
                if(node.right != null){
                    queue.add(node.right);
                }
            }
            if(depth%2 == 0){
                while(!stack.empty()){
                    pResult.get(depth-1).add(stack.peek());    //.peek()取栈顶元素但不删除它
                    stack.pop();    //栈内元素依次出栈，满足从右到左；.pop()会把栈顶元素删除
                }
            }
            depth++;
        }
        return pResult;

    }

}