通过从每组中获取一个值来计算对集

Question

我试图计算可能的成对数，这可以通过从两个集合中获取值来实现。没有人停下来Rest。我还尝试使用JavaSet实现它。但我陷入了逻辑，如何计算这种可能的组合。

问题示例：

Input:
Set 1: [0, 1, 4]
Set 2: [2, 3]
Set None: []

这里，可能的对组合是 [0，2] ， [0，3] ， [1，2] ， [1，3] ， [4，2] ， [4，3]

Output:
6 combination to choose pair by taking one value from each set

代码如下：

static Set<Integer> countryOne = new HashSet();
static Set<Integer> countryTwo = new HashSet();
static Set<Integer> countryNone = new HashSet();

static int journeyToMoon(int n, int[][] astronaut) {
    ///Separating diffrent country from input
    countryOne.add(astronaut[0][0]); countryOne.add(astronaut[0][1]);
    for(int i=1; i<astronaut.length; i++){
        boolean countryCheckFlag = false;
        for(int j=0; j<astronaut[i].length; j++){
            if(countryOne.contains(astronaut[i][j])){
                countryCheckFlag = true;
            }
        }
        if(countryCheckFlag){countryOne.add(astronaut[i][0]); countryOne.add(astronaut[i][1]);}
        else {countryTwo.add(astronaut[i][0]); countryTwo.add(astronaut[i][1]);}
    }

    ///Separating country which not present in input
    for(int i=0; i<n; i++){
        if(!countryOne.contains(i) && !countryTwo.contains(i))
            countryNone.add(i);
    }

    //Now i have two diffrent set

    return 0;
}

输入可能在某个级别上有所不同，如下所示

Input:
Set 1: [0, 2]
Set 2: []
Set None: [1, 3]

这里，可能的对组合是[0,1]、[0,3]、[2,1]、[2,3]，就像之前一样，但因为它是Set None，所以它会在Set None之间创建额外的集，如[1,3]

Output:
5 combination to choose pair by taking one value from each set

这里可能是答案，如，总组合= （集合 1 和集合 2 与乘积的可能组合）（集合 1 和集合 2 的可能组合，每个集合无）（集合无的所有可能组合具有不同值）

如果是，如何计算。输入范围将在1到10^5之间。谢谢。

Answer 1

要计算组合总数，只需使用

int a = set1.size() * set2.size();

要计算n元素对的总数，可以使用公式（n*n-n）/2

int x = setNone.size();
int b = (x * x - x) / 2