题解 | #牛群分层排列#

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 string字符串一维数组
     */
    public String[] levelOrder (TreeNode root) {
        // write code here
        List<String> ans = new ArrayList<>();
        levelOrder(root, 0, ans);
        return ans.toArray(new String[0]);
    }
    private void levelOrder(TreeNode root, int level, List<String> ans) {
        if (root == null) return;
        if (ans.size() == level) ans.add("");
        ans.set(level, ans.get(level) + root.val);
        levelOrder(root.left, level + 1, ans);
        levelOrder(root.right, level + 1, ans);
    }
}

• 基于二叉树的层序遍历求解
• 定义一个List集合ans，存放结果
• 调用层序遍历方法，边界值处理；当 ans 的size 等于 level时，初始化add空字符串
• 然后重置level索引位置的值：ans.set(level, ans.get(level) + root.val)
• 递归层序遍历
• 最后将ans转为String数组，返回结果