【剑指offer较难部分21】按之字形顺序打印二叉树（java）

题目描述

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

分析

一、队列
从题目中可以得出两个信息，一个是逐行打印，那么就用层次遍历，另一个由于是层次遍历，那么最先想到的就是队列。流程层次遍历队列流程就是先根节点入队，再只要队列不为空，就获取队列大小，训练进行出队，出队一个队列size就减1，size==0时，某一行就出队完毕。在层次遍历上加上ArrayList，并且设置一个flag，控制是从左往右还是从右往左。这是我看到这题思路。

实现代码如下：

import java.util.ArrayList;
import java.util.Queue;
import java.util.LinkedList;
public class Solution {
    
    
    public ArrayList<ArrayList<Integer> > Print(TreeNode pRoot) {
    
    
        //按行打印，即层次遍历，先左到右，再从右到左，可以设置一个flag
        ArrayList<ArrayList<Integer>> list = new ArrayList<>();
        if(pRoot == null){
    
    
            return list;
        }
        boolean flag = true;
        Queue<TreeNode> queue = new LinkedList<>();
        queue.offer(pRoot);
        while(!queue.isEmpty()){
    
    
            int count = queue.size();//每一行结点个数
            ArrayList<Integer> list_in = new ArrayList<>();
            while(count > 0){
    
    
                TreeNode cur = queue.poll();//出队
                count--;
                if(flag == true){
    
    
                    list_in.add(cur.val);//从左到右
                }else{
    
    
                    list_in.add(0,cur.val);//从右到左，即在index=0位置插入元素，之前元素往后移
                }
                if(cur.left != null){
    
    
                    queue.offer(cur.left);
                }
                if(cur.right != null){
    
    
                    queue.offer(cur.right);
                }
            }
            flag = !flag;
            list.add(list_in);
        }
        return list;
    }

}

二、两个栈
用两个栈解决是看到网上很多都用这种方法，就是利用栈的先进后出特性，可以实现从左到右，再从右到左。一个栈add一行后，又会到另一个栈再add一行。直到两个栈都为空，就执行结束。

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

public class Solution {
    
    
    public ArrayList<ArrayList<Integer> > Print(TreeNode pRoot) {
    
    
        ArrayList<ArrayList<Integer>> res = new ArrayList<>();
        if(pRoot==null)
            return res;
        Stack<TreeNode> s1 = new Stack<>();
        Stack<TreeNode> s2 = new Stack<>();
        s1.push(pRoot);
        TreeNode cur;
        while(!s1.isEmpty() || !s2.isEmpty()){
    
    
            ArrayList<Integer> list_in = new ArrayList<>();
            if(!s1.isEmpty()){
    
    
                while(!s1.isEmpty()){
    
    
                    cur = s1.pop();
                    list_in.add(cur.val);
                    if(cur.left != null)
                        s2.push(cur.left);
                    if(cur.right != null)
                        s2.push(cur.right);
                }
                res.add(list_in);
            }
            else{
    
    
                while(!s2.isEmpty()){
    
    
                    cur = s2.pop();
                    list_in.add(cur.val);
                    if(cur.right != null)
                        s1.push(cur.right);
                    if(cur.left != null)
                        s1.push(cur.left);
                }
                res.add(list_in);
            }
        }
        return res;
    }
}

个人觉得还是用队列简单