如何将字符串拆分为单词后均匀排列到n列二维数组中

原有代码问题梳理 #

你当前的代码存在以下不符合需求的问题：

• 列数通过平方根自动计算得出，未支持自定义输入n列的需求
• 采用逐行填充的逻辑，无法保证每列的单词数量均等
• 未处理单词总数无法被列数整除的场景，空值分布无规律

实现思路 #

我们按照以下规则保证单词均匀分布：

• 当单词总数可以被列数n整除时，每列单词数固定为 总单词数/n
• 当单词总数无法被n整除时，前 总单词数%n 列的单词数比其他列多1，保证所有列单词数差值不超过1
• 按单词原始顺序优先填充列，避免乱序

完整实现代码 #

import java.util.Arrays;

public class WordArranger {
    /**
     * 将输入字符串的单词均匀排列到指定列数的二维数组中
     * @param input 输入的空格分隔字符串
     * @param colCount 指定的列数n
     * @return 均匀分布后的二维数组
     */
    public static String[][] arrangeWords(String input, int colCount) {
        // 拆分单词
        String[] words = input.split(" ");
        int total = words.length;
        // 计算基础行数、多一行的列数
        int baseRow = total / colCount;
        int extraCol = total % colCount;

        // 初始化二维数组
        String[][] result = new String[baseRow + (extraCol > 0 ? 1 : 0)][colCount];
        int wordIndex = 0;

        // 逐列填充，保证每列单词数均匀
        for (int col = 0; col < colCount; col++) {
            // 当前列的行数：前extraCol列多一行
            int currentRowCount = baseRow + (col < extraCol ? 1 : 0);
            for (int row = 0; row < currentRowCount; row++) {
                if (wordIndex < total) {
                    result[row][col] = words[wordIndex++];
                }
            }
        }
        return result;
    }

    // 重载方法：自动计算近似正方形的列数，兼容原有逻辑
    public static String[][] arrangeWords(String input) {
        String[] words = input.split(" ");
        int colCount = (int) Math.ceil(Math.sqrt(words.length));
        return arrangeWords(input, colCount);
    }

    // 测试用例
    public static void main(String[] args) {
        String testInput = "我 是 一 个 测 试 字 符 串 共 有 十 一 个 单 词";
        // 测试指定列数
        int n = 4;
        String[][] res1 = arrangeWords(testInput, n);
        System.out.println("指定4列的排列结果：" + Arrays.deepToString(res1));
        // 测试自动算列数
        String[][] res2 = arrangeWords(testInput);
        System.out.println("自动计算列数的排列结果：" + Arrays.deepToString(res2));
    }
}

内容的提问来源于stack exchange，提问作者rickygrimes