Leetcode 1577 数的平方等于两数乘积的方法数

题目

给你两个整数数组 nums1 和 nums2 ，请你返回根据以下规则形成的三元组的数目（类型 1 和类型 2 ）：

• 类型 1：三元组 (i, j, k) ，如果$ nums1[i]2 == nums2[j] * nums2[k] 其中 0 <= i < nums1.length 且 0 <= j < k < nums2.length$
• 类型 2：三元组 (i, j, k) ，如果 $nums2[i]2 == nums1[j] * nums1[k] 其中 0 <= i < nums2.length 且 0 <= j < k < nums1.length$

示例 1：

输入：nums1 = [7,4], nums2 = [5,2,8,9]
输出：1
解释：类型 1：(1,1,2), nums1[1]^2 = nums2[1] * nums2[2] (4^2 = 2 * 8)

示例 2：

输入：nums1 = [1,1], nums2 = [1,1,1]
输出：9
解释：所有三元组都符合题目要求，因为 1^2 = 1 * 1
类型 1：(0,0,1), (0,0,2), (0,1,2), (1,0,1), (1,0,2), (1,1,2), nums1[i]^2 = nums2[j] * nums2[k]
类型 2：(0,0,1), (1,0,1), (2,0,1), nums2[i]^2 = nums1[j] * nums1[k]

示例 3：

输入：nums1 = [7,7,8,3], nums2 = [1,2,9,7]
输出：2
解释：有两个符合题目要求的三元组
类型 1：(3,0,2), nums1[3]^2 = nums2[0] * nums2[2]
类型 2：(3,0,1), nums2[3]^2 = nums1[0] * nums1[1]

示例 4：

输入：nums1 = [4,7,9,11,23], nums2 = [3,5,1024,12,18]
输出：0
解释：不存在符合题目要求的三元组

提示：

1 <= nums1.length, nums2.length <= 1000
1 <= nums1[i], nums2[i] <= 10^5

思路

利用哈希表。

首先对第一个数组的每一个数取平方，然后插入到哈希表中，如果已经存在那么就在原来大小上+1.

然后遍历第二个数组，取两个数的乘积去哈希表里寻找，直接加上哈希表中的数值。

这个数值是有意义的，比如上面给出的示例2，

示例 2：

输入：nums1 = [1,1], nums2 = [1,1,1]
输出：9
解释：所有三元组都符合题目要求，因为 1^2 = 1 * 1
类型 1：(0,0,1), (0,0,2), (0,1,2), (1,0,1), (1,0,2), (1,1,2), nums1[i]^2 = nums2[j] * nums2[k]
类型 2：(0,0,1), (1,0,1), (2,0,1), nums2[i]^2 = nums1[j] * nums1[k]

nums1里只会得到两个结果是1的值，如果后面遍历后查找哈希表，会只得到“能找到这样的匹配”的结果。但是如果我们让数字有意义，存放nums1中平方得到该值的个数，那么后面遍历时能得到的就不仅仅是能找到这样的匹配了，而是“这样的匹配有n个”，把n加到结果上即可。

注意：

• int相乘可能会超过int的范围，这时候要用long，而且要主动转换类型，否则int*int>int范围就会直接报错了。
• 注意翻转两个数组的顺序再找一遍。

代码与结果

class Solution {
public:
    int numTriplets(vector<int>& nums1, vector<int>& nums2) {
        return hashAndFind(nums1,nums2) + hashAndFind(nums2,nums1);
    }
private:
    int hashAndFind(vector<int>& nums1, vector<int>& nums2){
        int res = 0;
        map<long, int> mp;
        for(auto& num : nums1){
            ++mp[(long)num*num];
        }
        int n = nums2.size();
        for(int i = 0; i< n-1; ++i){
            for(int j=i+1; j<n; ++j){
                res += mp[(long)nums2[i]*nums2[j]];
            }
        }
        return res;
    }
};

结果（大概是没啥人做这个题，所以排名靠前