題目

給你兩個整數陣列 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. 預處理得出兩個陣列的所有乘積，用map存起來每種結果出現的次數
3. 再次遍歷求冪，即可得出結果

程式碼

class Solution {
public:
    int numTriplets(vector<int>& nums1, vector<int>& nums2) {
        int res=0;
        unordered_map<long long,int> pro1,pro2;
        for(int i=0;i<nums1.size();i++){
            for(int j=i+1;j<nums1.size();j++){
                long pro=(long)nums1[i]*(long)nums1[j];
                pro1[pro]++;
            }
        }
        for(int i=0;i<nums2.size();i++){
            for(int j=i+1;j<nums2.size();j++){
                long pro=(long)nums2[i]*(long)nums2[j];
                pro2[pro]++;
            }
        }
        for(int i=0;i<nums1.size();i++){
            long check=(long)nums1[i]*(long)nums1[i];
            res+=pro2[check];
        }
        
        for(int i=0;i<nums2.size();i++){
            long check=(long)nums2[i]*(long)nums2[i];
            res+=pro1[check];
        }
        
        return res;
    }
};