标签：medium python res len 三元组 range 81 nums1 nums2

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

题目描述：
给你两个整数数组 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
示例 ：
输入：nums1 = [7,4], nums2 = [5,2,8,9]
输出：1
提示：

• 1 <= nums1.length, nums2.length <= 1000
• 1 <= nums1[i], nums2[i] <= 10⁵

解法

哈希表存储两个数组的平方，然后两次遍历另一个数组的两元素乘积。

代码

class Solution:
    def numTriplets(self, nums1: List[int], nums2: List[int]) -> int:
        nums12pos = Counter([x**2 for x in nums1])
        nums22pos = Counter([x**2 for x in nums2])
        res = 0
        for i in range(len(nums2)):
            for j in range(i + 1, len(nums2)):
                res += nums12pos[nums2[i] * nums2[j]]
        for i in range(len(nums1)):
            for j in range(i + 1, len(nums1)):
                res += nums22pos[nums1[i] * nums1[j]]
        return res
      

测试结果

执行用时：360 ms, 在所有 Python3 提交中击败了 20.69% 的用户
内存消耗：15 MB, 在所有 Python3 提交中击败了 82.76% 的用户

说明

算法题来源：力扣(LeetCode)

标签：medium,python,res,len,三元组,range,81,nums1,nums2
来源： https://blog.csdn.net/qq_40137656/article/details/120790664

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。