题解 | #牛群Z字型排列#

题目考察的知识点

和上一道题目相似，考察二叉树的层序遍历

题目解答方法的文字分析

用一个标记位height来表示高度，借助列表逆序操作将对应层的序列逆序，最后的数组转换注意以下就可以了。细节可以看看注释

本题解析所用的编程语言

使用Java解答

完整且正确的编程代码

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 {
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     * 
     * @param root TreeNode类 
     * @return int整型二维数组
     */
    int height = 1;
    public int[][] ZLevelOrder (TreeNode root) {
        // write code here
        List<List<Integer>> ret = new ArrayList<List<Integer>>();
        if (root == null) {
            return new int[0][0];
        }
        Queue<TreeNode> queue = new LinkedList<TreeNode>(); //队列 数据类型是TreeNode
        queue.offer(root); //加入节点
        while (!queue.isEmpty()) {
            List<Integer> list = new ArrayList<Integer>();
            //String str = "";
            int currentLevelSize = queue.size();
            for (int i = 1; i <= currentLevelSize; ++i) {
                TreeNode node = queue.poll();
                list.add(node.val);
                if (node.left != null) {
                    queue.offer(node.left);
                }
                if (node.right != null) {
                    queue.offer(node.right);
                }
            }
            if(height%2==0) Collections.reverse(list); //根据高度判断是否逆序
            height++;
            ret.add(list);
        }
        int[][] res = new int[ret.size()][];
        for(int i=0; i<ret.size(); i++){
            int[] arr = new int[ret.get(i).size()]; //注意
            for(int j=0; j<ret.get(i).size();j++){
                arr[j] = ret.get(i).get(j);
            }
            res[i] = arr;
        }
        return res;
    }
}