剑指offer_ 按之字形顺序打印二叉树

题目描述

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

和上题差不多，加上一个反转list就行了，设置一个flag标志位，每循环一轮就反置。可以使用Collections工具类的reverse()方法。

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 {
    ArrayList<ArrayList<Integer> > ans= new ArrayList<>();
    ArrayList<TreeNode> queue = new ArrayList<>();
    boolean flag = false;
    public ArrayList<ArrayList<Integer> > Print(TreeNode pRoot) {
        if(pRoot ==null) return ans;
        queue.add(pRoot);
        while(!queue.isEmpty()){
            int count = queue.size();
            ArrayList<Integer> arr= new ArrayList<>();
            while(count-->0){
            TreeNode temp = queue.remove(0);
            arr.add(temp.val);
            if(temp.left!=null){queue.add(temp.left);}
            if(temp.right!=null){queue.add(temp.right);}
            }
            if(flag){
                Collections.reverse(arr);
            }
            flag = !flag;
            ans.add(arr);
        }
            return ans;
    }

}