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

牛群Z字型排列

https://www.nowcoder.com/practice/50ddaf50c6954600a9f1dbb099d6f388

题目考察的知识点是:

本题主要考察二叉树、循环。

题目解答方法的文字分析:

该题求我们按层数的奇偶来决定每一层的输出顺序。规定二叉树的根节点为第 0 层,如果当前层数是偶数,从左至右输出当前层的节点值,否则,从右至左输出当前层的节点值。因此我们通过层次遍历维护每一层的输出数组。同时维护一个isleft,当isleft为falses时,将得到的正序输出数组逆转成反序;为true则不需要做反转处理。然后用一个list将所有的输出数组装载,算法最后将它为int[][]类型

本题解析所用的编程语言:

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整型二维数组
     */
    public int[][] ZLevelOrder (TreeNode root) {
        // write code here
        ArrayList<ArrayList<Integer>> res = new ArrayList<>();
        if (root == null) {
            return new int[0][0];
        }
        Queue<TreeNode> queue = new LinkedList<>();
        queue.add(root);
        ArrayList<Integer> list = new ArrayList<>();
        list.add(root.val);
        res.add(list);
        boolean order = false;
        while (queue.size() != 0) {
            int n = queue.size();
            ArrayList<Integer> list1 = new ArrayList<>();
            for (int i = 0; i < n; i++) {
                TreeNode cur = queue.poll();
                if (cur.left != null) {
                    queue.add(cur.left);
                    list1.add(cur.left.val);
                }
                if (cur.right != null) {
                    queue.add(cur.right);
                    list1.add(cur.right.val);
                }
            }
            if (!order) {
                Collections.reverse(list1);
            }
            if (list1.size() != 0) {
                res.add(list1);
            }
            order = !order;
        }
        int[][] ans = new int[res.size()][];
        for (int i = 0; i < res.size(); i++) {
            int[] arr = new int[res.get(i).size()];
            for (int j = 0; j < res.get(i).size(); j++) {
                arr[j] = res.get(i).get(j);
            }
            ans[i] = arr;
        }
        return ans;
    }
}

#题解#
全部评论

相关推荐

1 收藏 评论
分享
牛客网
牛客企业服务