java - 将列表分成组

Question

我有一个字符串列表，我想知道是否有一种简单的方法可以将元素加入大小为 n 的组中，如果小于 n，最后一个元素将被自动处理。例如，

String[] strings = {"A", "man", "a", "plan", "a", "canal", "Panama"};

n = 3 的预期输出将是以下字符串列表。请记住，我可以自己使用一些令人筋疲力尽的迭代来完成此操作，但我正在寻找简单的东西。也许是标准库内置的东西。

String[] output = {"A man a", "plan a canal", "Panama"};

作为比较，我想演示一下 Python 中任务的简单性。我正在寻找一个同样简单的 Java 解决方案。

[' '.join(strings[i:i+3]) for i in range(0, len(strings), 3)]

Answer 1

• Guava - Google 的 Java 实用程序库，有 partition Lists.partition(list, size) 方法可用于将列表分成 3 组
• Java 8 有 String.join(separator, iterable)，或者 Guava 的 Joiner.join()
• 像列表这样的集合通常比原始数组更适合使用......

示例，假设您无权访问 Guava。

public static void main(String[] args) {
    String[] strings = { "A", "man", "a", "plan", "a", "canal", "Panama" };

    List<String> output = new ArrayList<String>();
    for (List<String> partition : partition(Arrays.asList(strings), 3)) {
        output.add(String.join(" ", partition));
    }
    System.out.println(Arrays.toString(output.toArray()));
}

private static <T> List<List<T>> partition(List<T> input, int size) {
    List<List<T>> lists = new ArrayList<List<T>>();
    for (int i = 0; i < input.size(); i += size) {
        lists.add(input.subList(i, Math.min(input.size(), i + size)));
    }
    return lists;
}