java - 可变长度数组的矩阵的组合

Question

我必须获得这种矩阵的所有可能的组合:

String[][] matrix = {
   {"AA-123", "AA-124", "AA-125", "AA-126"},
   {"BB-12", "BB-13"},
   {"CC-1"},
};

毕竟那是最终的实现。它是用 Java 编写的，但该语言可能无关紧要:

long nComb = 1;
for (int iMatr = 0; iMatr < matrix.length; iMatr++)
   nComb *= matrix[iMatr].length;

for (int iComb = 0; iComb < nComb; iComb++) {
   System.out.print("|");

   long nSec = 1;
   for (int iSec = 0; iSec < matrix.length; iSec++) {
       String[] sec = matrix[iSec];

       for (int iAtom = 0; iAtom < sec.length; iAtom++) {

           if (iAtom == ((iComb / nSec) % sec.length))
               System.out.print(1);
           else
               System.out.print(0);
       }

       nSec *= sec.length;
       System.out.print("|");
   }

   System.out.println();
}

我必须将我的逻辑应用到打印 1 或 0 的 if 上。我需要知道数组组合的当前元素(索引)是什么。预期结果:

|1000|10|1|
|0100|10|1|
|0010|10|1|
|0001|10|1|
|1000|01|1|
|0100|01|1|
|0010|01|1|
|0001|01|1|

问候。

编辑:

我在数组迭代中使用另一个变量找到了可能的答案:nSec。它通过迭代增加数组的长度，在最后一次迭代时达到 nComb 的值。

Answer 1

我相信你在这里追求的是多个集合的笛卡尔积，并且Java中的许多集合库已经支持它。

我个人建议使用 Guava ( https://code.google.com/p/guava-libraries/ )，它允许您按如下方式定义您的问题(数组和集合之间的转换我将作为练习省略:):

import com.google.common.collect.Sets;
import java.util.List;
import java.util.Set;

public class CartesianProduct {

    public static void main(String[] args) {
        Set<List<String>> merged = Sets.cartesianProduct(
                Sets.newHashSet("AA-123", "AA-124", "AA-125", "AA-126"),
                Sets.newHashSet("BB-12", "BB-13"),
                Sets.newHashSet("CC-1")
        );
        System.out.println("Size: " + merged.size());
        System.out.println("Content: " + merged);
    }

}

通过执行此代码，您将得到以下结果:

Size: 8
Content: [
   [AA-125, BB-13, CC-1], 
   [AA-125, BB-12, CC-1], 
   [AA-124, BB-13, CC-1], 
   [AA-124, BB-12, CC-1], 
   [AA-123, BB-13, CC-1], 
   [AA-123, BB-12, CC-1], 
   [AA-126, BB-13, CC-1], 
   [AA-126, BB-12, CC-1]
]

然后，您可以按照您需要的方式处理、排序和格式化输出(打印时另一个 Guava 类 Joiner 可能会派上用场)。