leetcode1577. 数的平方等于两数乘积的方法数（Python3、c++）

leetcode1577. 数的平方等于两数乘积的方法数

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

---

方法：哈希表+模拟

思路：

本题可以很容易的使用哈希表（python中的字典）来进行处理。

我们先使用两个哈希表square1和square2，square1ms1中每个数的平方值以及对应的出现次数（键为平方值，值为出现次数），square2存放nums2中每个数的平方值以及对应的出现次数。这里面需要注意的是，对于c++，需要考虑整数溢出，使用long long类型。

然后，我们分别两层循环遍历nums1和nums2数组，在nums1中，找到所有的(j,k)数对，计算两者的乘积在square2中的次数，更新答案，同理在nums2中，找到所有的(j,k)数对，计算两者的乘积在square1中的次数，更新答案。

最后返回答案即可。

代码:

Python3:

class Solution:
    def numTriplets(self, nums1: List[int], nums2: List[int]) -> int:
        n1 = len(nums1)
        n2 = len(nums2)
        # square1存放nums1中每个数的平方值对应的出现次数
        # square2存放nums2中每个数的平方值对应的出现次数