Java картезианская коллекция произведений

Question

Мне нужно сделать декартово произведение из 3 списков. Вот код

 XList<T> combine() {
    XList<T> returnList = new XList<>();

    List<T> list = new ArrayList(this.coll);
    List<T> a = (List<T>) list.get(0);
    List<T> b = (List<T>) list.get(1);
    List<T> c = (List<T>) list.get(2);

    //working
    //List<XList<T>> ll2 = a.stream().flatMap(ai -> b.stream().map(bi -> new XList<>(Arrays.asList(ai, bi)))).collect(Collectors.toList());

    // not working
    List<XList<T>> ll3 = a.stream().flatMap(ai -> b.stream().flatMap(bi -> c.stream().map(ci -> new XList<>(Arrays.asList(ai, bi, ci))))).collect(Collectors.toList());


    returnList.coll = (Collection<T>) ll3;

    return returnList;
}

Когда я пытаюсь сделать декартово произведение из списка a и b, оно работает, но если я хочу сделать это с a,b и c, оно возвращает пустой список. Что это не так? Спасибо за ответ.

Answer 1

Гораздо лучше использовать существующие решения, такие как гуава

В качестве альтернативы вы можете использовать эту библиотеку

Answer 2

Я думаю, что я не могу использовать некоторые другие библиотеки. Если у кого-то есть идея, я был бы очень благодарен.

Answer 3

Вы можете сгенерировать декартово произведение нескольких коллекций, используя три вложенных цикла for . Последовательное добавление данных из входящих коллекций к промежуточному результату и получение конечного результата схематично выглядит так:

      res0: [[]]
col1: [1,2,3]
----
res1: [[1],[2],[3]]
col2: [4,5,6]
----
res2: [[1,4],[1,5],[1,6],[2,4],[2,5]...,[3,6]]
col3: [7,8,9]
----
res3: [[1,4,7],[1,4,8],[1,4,9],[1,5,7],[1,5,8]...,[3,6,9]]

Попробуйте онлайн!

      /**
 * @param cols an arbitrary number of collections
 * @param <T>  the type of the elements
 * @return the Cartesian product
 */
@SafeVarargs
public static <T> List<List<T>> cartesianProduct(Collection<T>... cols) {
    // check if incoming data is not null
    if (cols == null) return Collections.emptyList();
    // Cartesian product, intermediate result
    List<List<T>> cp = Collections.singletonList(Collections.emptyList());
    // iterate through the incoming collections
    for (Collection<T> col : cols) {
        // non-null and non-empty collections
        if (col == null || col.size() == 0) continue;
        // intermediate result for next iteration
        List<List<T>> next = new ArrayList<>();
        // rows of current intermediate result
        for (List<T> row : cp) {
            // elements of current list
            for (T el : col) {
                // new row for next intermediate result
                List<T> nRow = new ArrayList<>(row);
                nRow.add(el);
                next.add(nRow);
            }
        }
        // pass to next iteration
        cp = next;
    }
    // Cartesian product, final result
    return cp;
}

      public static void main(String[] args) {
    List<Integer> list1 = Arrays.asList(1, 2, 3);
    List<Integer> list2 = Arrays.asList(4, 5, 6);
    List<Integer> list3 = Arrays.asList(7, 8, 9);

    List<List<Integer>> cp = cartesianProduct(list1, list2, list3);
    // column-wise output
    int rows = 9;
    for (int i = 0; i < rows; i++) {
        for (int j = 0; j < cp.size(); j++)
            System.out.print(j % rows == i ? cp.get(j) + " " : "");
        System.out.println();
    }
}

Выход:

      [1, 4, 7] [2, 4, 7] [3, 4, 7] 
[1, 4, 8] [2, 4, 8] [3, 4, 8] 
[1, 4, 9] [2, 4, 9] [3, 4, 9] 
[1, 5, 7] [2, 5, 7] [3, 5, 7] 
[1, 5, 8] [2, 5, 8] [3, 5, 8] 
[1, 5, 9] [2, 5, 9] [3, 5, 9] 
[1, 6, 7] [2, 6, 7] [3, 6, 7] 
[1, 6, 8] [2, 6, 8] [3, 6, 8] 
[1, 6, 9] [2, 6, 9] [3, 6, 9] 

---

^{См. Также: Как получить декартово произведение из нескольких списков?}