题解 | #牛群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) {
if (root == null) {
return new int[0][0];
}
List<int[]> list = new LinkedList<>();
Queue<TreeNode> nodeQueue = new ArrayDeque<TreeNode>();
nodeQueue.offer(root);
boolean isleft=true;
while(!nodeQueue.isEmpty()) {
int[] ret = new int[nodeQueue.size()];
int p = 0;
int size = nodeQueue.size();
while(size-- >0){
TreeNode cur = nodeQueue.poll();
ret[p++] = cur.val;
if(cur.left!=null){
nodeQueue.add(cur.left);
}
if(cur.right!=null){
nodeQueue.add(cur.right);
}
}
if(!isleft){
int left=0;
int right=ret.length-1;
while(left<right){
Integer tem = ret[left];
ret[left] = ret[right];
ret[right] = tem;
left++;
right--;
}
}
list.add(ret);
isleft=!isleft;
}
int[][] ans = new int[list.size()][];
int t=0;
for(int[] i:list){
ans[t++] = i;
}
return ans;
}
}