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

这段代码实现了按照Z字形顺序遍历二叉树，并返回每层节点值的二维整数数组。

首先，我们定义了一个树节点类TreeNode，它包含一个整数值val以及左右子节点的引用。

然后，在Solution类中，我们声明了一个公共的ZLevelOrder方法，该方法接受一个树节点root作为参数，并返回按照Z字形顺序遍历得到的每层节点值的二维整数数组。

在ZLevelOrder方法中，我们首先进行了一个特殊情况的判断，即如果树节点root为null，则返回一个空的二维数组。

接下来，我们使用一个队列来辅助进行层序遍历。同时，我们使用一个布尔变量leftToRight来表示当前层的遍历方向，初始值为true，表示从左到右。

然后，我们通过循环遍历每一层的节点。在每次循环开始前，我们记录当前队列的大小，即当前层的节点数量。

在每个循环中，我们首先创建一个列表levelNodes来保存当前层的节点值。然后，根据leftToRight的值来决定是从左到右还是从右到左添加节点值到levelNodes中。

接下来，我们遍历当前层的所有节点，将其左右子节点加入队列中。

循环结束后，我们切换下一层的遍历方向，将leftToRight取反。然后，将当前层的节点值列表levelNodes添加到结果列表result中。

最后，我们将结果列表result转换为二维整数数组，并返回结果。

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整型二维数组
     */
    public int[][] ZLevelOrder (TreeNode root) {
        if (root == null) {
            return new int[0][];
        }

        List<List<Integer>> result = new ArrayList<>();
        Queue<TreeNode> queue = new LinkedList<>();
        boolean leftToRight = true;

        queue.offer(root); // 将根节点入队

        while (!queue.isEmpty()) {
            int levelSize = queue.size();
            List<Integer> levelNodes = new ArrayList<>();

            // 遍历当前层的所有节点
            for (int i = 0; i < levelSize; i++) {
                TreeNode node = queue.poll();

                // 根据遍历方向添加节点值到当前层列表
                if (leftToRight) {
                    levelNodes.add(node.val);
                } else {
                    levelNodes.add(0, node.val);
                }

                // 将当前节点的左右子节点入队
                if (node.left != null) {
                    queue.offer(node.left);
                }
                if (node.right != null) {
                    queue.offer(node.right);
                }
            }

            leftToRight = !leftToRight; // 切换下一层的遍历方向
            result.add(levelNodes); // 将当前层列表添加到结果列表中
        }

        // 将结果转换为二维数组
        int[][] zLevelOrder = new int[result.size()][];
        for (int i = 0; i < result.size(); i++) {
            List<Integer> levelNodes = result.get(i);
            zLevelOrder[i] = new int[levelNodes.size()];
            for (int j = 0; j < levelNodes.size(); j++) {
                zLevelOrder[i][j] = levelNodes.get(j);
            }
        }

        return zLevelOrder;
    }
}