java - 生成不同集合的组合

Question

我有同一个对象的 3 个集合，但每个对象都有一个类型属性，例如枚举。

每个集合都有具有相同类型属性的对象。因此，集合 1 除了类型属性外，所有对象属性都不同，所有其他集合都相同。

现在我想要拥有这 3 个集合的所有组合，例如结果集需要有 1 项 set1、2 项 set2 和 1 项 set3(这是可配置的)。

所以我尝试的是将这 3 个集合合并为一个集合，并使用 google guava Sets.combinations(bigSet, 4) 来获取这些大小为 4 的集合的所有组合。之后，我会过滤掉那些不存在的集合。不满足这些规则。

我担心这不是最有效的方法。因为首先该方法会生成比我需要的更多组合，然后我需要过滤 3 次(每组 1 次)以删除不具有正确数量的类型对象的组。因此，如果将来我需要额外的一套，则需要过滤 4 次。

是否有更好的方法来生成这些集合？

虚拟类:

public class Dummy {
    String name;
    Type type;

    public Dummy(String name, Type type) {
        this.name = name;
        this.type = type;
    }

    @Override
    public String toString() {
        return "Dummy{" +
                "name='" + name + '\'' +
                ", type=" + type +
                '}';
    }
}

测试代码:

    Dummy d1 = new Dummy("Rik", Type.TYPE1);

    Dummy d2 = new Dummy("John", Type.TYPE2);
    Dummy d3 = new Dummy("Bart", Type.TYPE2);

    Dummy d4 = new Dummy("Elisabeth", Type.TYPE3);
    Dummy d5 = new Dummy("Annie", Type.TYPE3);

    Set<Dummy> dummySet1 = ImmutableSet.of(d1);
    Set<Dummy> dummySet2 = ImmutableSet.of(d2, d3);
    Set<Dummy> dummySet3 = ImmutableSet.of(d4, d5);;

    Set<Dummy> result = new HashSet<Dummy>();
    result.addAll(dummySet1);
    result.addAll(dummySet2);
    result.addAll(dummySet3);
    System.out.println(result);

    Set<Set<Dummy>> combinations = Sets.combinations(result, 4);

    Set<Set<Dummy>> endResult = new HashSet<>();
    for (Set<Dummy> set : combinations) {
        long type1Count = set.stream().filter(s -> s.type == Type.TYPE1).count();
        long type2Count = set.stream().filter(s -> s.type == Type.TYPE2).count();
        long type3Count = set.stream().filter(s -> s.type == Type.TYPE3).count();

        if (type1Count == 1 && type2Count == 2 && type3Count == 1) {
            endResult.add(set);
        }
    }

    for (Set<Dummy> set : endResult) {
        System.out.println(set);
    }

Answer 1

使用它，因为它不会生成无效的组合:

Sets.cartesianProduct(
   Sets.combinations(Set1, set1Count), 
   Sets.combinations(Set2, set2Count), 
   Sets.combinations(Set3, set3Count));