leetcode 5508. 数的平方等于两数乘积的方法数（C++）

给你两个整数数组 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

C++

class Solution {
public:
    int numTriplets(vector& nums1, vector& nums2) 
    {
        long res=0;
        unordered_map tmp1;
        unordered_map tmp2;
        int m=nums1.size();
        int n=nums2.size();
        for(int i=0;i vec;
            for(auto iv:tmp2)
            {
                int v1=iv.first;
                if(vec.find(v1)==vec.end())
                {
                    int v2=val/v1;
                    if(val==(long)v1*(long)v2 && tmp2.find(v2)!=tmp2.end())
                    {
                        if(v1==v2)
                        {
                            int num1=iv.second;
                            res+=(long)num*(long)num1*(long)(num1-1)/2;
                        }
                        else
                        {
                            int num1=iv.second;
                            int num2=tmp2[v2];
                            res+=(long)num*(long)num1*(long)num2;
                        }
                        vec[v2]=1;
                    }
                }
            }
        }
        for(auto it:tmp2)
        {
            long val=(long)it.first*(long)it.first;
            int num=it.second;
            map vec;
            for(auto iv:tmp1)
            {
                int v1=iv.first;
                if(vec.find(v1)==vec.end())
                {
                    int v2=val/v1;
                    if(val==(long)v1*(long)v2 && tmp1.find(v2)!=tmp1.end())
                    {
                        if(v1==v2)
                        {
                            int num1=iv.second;
                            res+=(long)num*(long)num1*(long)(num1-1)/2;
                        }
                        else
                        {
                            int num1=iv.second;
                            int num2=tmp1[v2];
                            res+=(long)num*(long)num1*(long)num2;
                        }
                        vec[v2]=1;
                    }
                }
            }
        }
        return res;
    }
};