我们可以按如下方式构建解决方案。首先请注意,列表大小不同的情况在现代语言中从来都不是问题,因为今天几乎所有语言都支持可变大小的列表。更棘手的部分是列表数量不同时。
这是固定数量列表的 Python 案例:
lista = ["a1","a2"]
listb = ["b1","b2","b3"]
listc = ["c1","c2"]
[(a,b,c) for a in lista for b in listb for c in listc]
对于 Java,做同样的事情。您将需要三个嵌套的 for 循环:
List<String> lista = Arrays.asList("a1","a2");
List<String> listb = Arrays.asList("b1","b2","b3");
List<String> listc = Arrays.asList("c1","c2");
List<List<String>> result = new ArrayList<List<String>>();
for (String a: lista) {
for (String b: listb) {
for (String c: listc) {
result.add(Arrays.asList(a, b, c));
}
}
}
System.out.println(result);
这产生
[[a1, b1, c1], [a1, b1, c2], [a1, b2, c1], [a1, b2, c2], [a1, b3, c1], [a1, b3, c2], [a2, b1, c1], [a2, b1, c2], [a2, b2, c1], [a2, b2, c2], [a2, b3, c1], [a2, b3, c2]]
现在的诀窍是如何执行不确定数量的 for 循环?一种方法是使用递归。
从 0 个列表(甚至 1 个)的基本情况开始,然后问自己:当我添加一个新列表时,结果如何变化?然后你可以组合一个很好的小递归公式。您需要这方面的帮助吗?
(顺便说一句:在 Python 中,我们有itertools
这些东西。我不知道有任何 Java 类似物,但一些谷歌搜索可能会发现一些东西。)
好的,让我们为无限数量的列表开发递归公式。如果您只有一个列表,请说
a = ["a1", "a2"]
那么结果是[["a1"],["a2"]]
。但现在让我们添加另一个列表:
a = ["b1", "b2"]
现在的答案是什么?
[[a1, b1], [a1, b2], [a2, b1], [a2, b2]]
我们是怎么得到的?答案是我们获取新列表的每个元素并将其附加到结果中的每个元素。每次我们添加一个新列表时,我们都会这样做。假设现在我们添加["c1","c2","c3"]
. 你看到我们会得到什么吗?如果是这样,您现在应该能够定义递归步骤!
完整代码
好吧,我无法抗拒!干得好。
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
/**
* An application requested by a SO user.
*/
public class ProductExample {
/**
* Returns a list containing all possible products of a list of lists of strings.
*/
public static List<List<String>> product(List<List<String>> lists) {
List<List<String>> result = new ArrayList<List<String>>();
if (lists.isEmpty()) {
result.add(new ArrayList<String>());
return result;
}
List<List<String>> partial = product(lists.subList(0, lists.size() - 1));
for (List<String> list: partial) {
for (String s: lists.get(lists.size() - 1)) {
List<String> listCopy = new ArrayList<String>(list);
listCopy.add(s);
result.add(listCopy);
}
}
return result;
}
public static void main(String[] args) {
System.out.println(product(
Arrays.asList(
Arrays.asList("a1", "a2"),
Arrays.asList("b1", "b2", "b3"),
Arrays.asList("c1", "c2"),
Arrays.asList("d1", "d2"))));
}
}
Rant:这在 Python 中要容易得多。或者红宝石。:)