题解 | #牛的层级关系#

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[][] levelOrder (TreeNode root) {
        Queue<TreeNode> treeNodeQueue = new LinkedList<>();
        if(root !=null){
            treeNodeQueue.add(root);
        }

        ArrayList<ArrayList<Integer>> arrayLists = new ArrayList<>();
        while(!treeNodeQueue.isEmpty()){
            int size = treeNodeQueue.size();
            ArrayList<Integer> arrayList = new ArrayList<>();
            for (int i = 0; i < size; i++) {
                TreeNode treeNode = treeNodeQueue.poll();
                arrayList.add(treeNode.val);
                if(treeNode.left!=null){
                    treeNodeQueue.add(treeNode.left);
                }
                if(treeNode.right!=null){
                    treeNodeQueue.add(treeNode.right);
                }
            }
            arrayLists.add(arrayList);
        }
        int [][] result = new int[arrayLists.size()][];
        for (int i = 0; i < arrayLists.size(); i++) {
            int [] temp = new int[arrayLists.get(i).size()];
            for (int j = 0; j < arrayLists.get(i).size(); j++) {
                temp[j] = arrayLists.get(i).get(j);
            }
            result[i] = temp;
        }
        return result;
    }
}

本题知识点分析：

1.二叉树层序遍历

2.DFS深度优先搜索

3.集合存取

4.集合转数组

本题解题思路分析：

1.简单的层序遍历，创建集合存储结果，因为不知道每层的数量

2.先加根节点加入队列，然后当队列不为空则是循环条件，获取队列的大小，for循环队列大小，取出treeNode,节点值加入临时集合，如果左右节点不为null,加入到队列中。

3.最后集合转数组返回

本题使用编程语言: Java

如果你觉得本篇文章对你有帮助的话，可以点个赞支持一下，感谢~