LeetCode 1726. Tuple with Same Product

🔗 https://leetcode.com/problems/tuple-with-same-product

题目

• 给无重复数字的数组，返回形成 a * b = c * d 的个数，其中 abcd 各不相同

思路

• 组合数学题
• 枚举统计所有数字两两相乘的 product，并进行计数
• 对于一组满足 a * b = c * d 的数字，可以通过排列组合形成 8 个可能性
• 对于某个 product 值的 count，可以形成的排列组合数是 count * （count - 1）/2 * 8

代码

class Solution {
public:
    int tupleSameProduct(vector<int>& nums) {
        int ans = 0;
        map<int, int> mp;
        for (int i = 0; i < nums.size() - 1; i++) {
            for (int j = i + 1; j < nums.size(); j++) {
                int tmp = nums[i] * nums[j];
                mp[tmp]++;
                int count = mp[tmp];
                if (count >= 2) {
                    ans += count * (count-1) / 2;
                    ans -= (count-1)*(count-2) /2;
                 }
            }
        }

        ans *= 8;
        return ans;
        
    }
};