给你一个由 不同 正整数组成的数组 nums
,请你返回满足 a * b = c * d
的元组 (a, b, c, d)
的数量。其中 a
、b
、c
和 d
都是 nums
中的元素,且 a != b != c != d
。
示例 1:
输入:nums = [2,3,4,6] 输出:8 解释:存在 8 个满足题意的元组: (2,6,3,4) , (2,6,4,3) , (6,2,3,4) , (6,2,4,3) (3,4,2,6) , (4,3,2,6) , (3,4,6,2) , (4,3,6,2)
示例 2:
输入:nums = [1,2,4,5,10] 输出:16 解释:存在 16 个满足题意的元组: (1,10,2,5) , (1,10,5,2) , (10,1,2,5) , (10,1,5,2) (2,5,1,10) , (2,5,10,1) , (5,2,1,10) , (5,2,10,1) (2,10,4,5) , (2,10,5,4) , (10,2,4,5) , (10,2,5,4) (4,5,2,10) , (4,5,10,2) , (5,4,2,10) , (5,4,10,2)
提示:
-
1 <= nums.length <= 1000
-
1 <= nums[i] <= 104
-
nums
中的所有元素 互不相同
class Solution { public: int tupleSameProduct(vector<int>& nums) { unordered_map<int,int>mp; int ans=0; for(int i=0;i<nums.size();i++){ for(int j=i+1;j<nums.size();j++){ mp[nums[i]*nums[j]]++; } } for(auto &[k,v]:mp){ ans+=v*(v-1); } return ans*8/2; } }; /* 今日总结: 1.auto可以遍历map,方式for(auto &[k,v]:mp) 2.组合数C(n,m),从n个不同元素中取出m(m≤n)个元素的所有组合的个数,C(n,m)=n!/(m!*(n-m)!) */