LeetCode 1726. 同积元组（map+组合数学）

题意：

给你一个由 不同 正整数组成的数组 nums ，请你返回满足 a * b = c * d 的元组 (a, b, c, d) 的数量。
其中 a、b、c 和 d 都是 nums 中的元素，且 a != b != c != d 。

数据范围:
1 <= nums.length <= 1000
1 <= nums[i] <= 104
nums 中的所有元素 互不相同

解法：

枚举a[i]*a[j],其中i!=j,
用map存下a[i]*a[j]=p的对数.

对于每个p,设有x个有序数对可以组成p,
从中取出两对分别作为a*b和c*d,方案数为C(x,2)=x*(x-1)/2,
同时a,b和c,d可以交换位置,(a,b)和(c,d)也可以,
因此方案数还需要乘上2*2*2=8.

code：

class Solution {
public:
    int tupleSameProduct(vector<int>& a) {
        int n=a.size();
        int ans=0;
        map<int,int>mark;
        for(int i=0;i<n;i++){
            for(int j=0;j<i;j++){
                mark[a[i]*a[j]]++;
            }
        }
        for(auto i:mark){
            int x=i.second;
            //有x个有序数对满足条件
            ans+=x*(x-1)/2*8;
        }
        return ans;
    }
};

---