生成不同集合的组合

Question

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

每个集合都有相同类型的对象属性。所以设置 1 除了类型属性之外，所有对象属性都不同，所有其他设置都相同。

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

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

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

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

虚拟 class:

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));

生成不同集合的组合

Generate combinations of different sets

java

guava