哈希表题目：数的平方等于两数乘积的方法数

题目

标题和出处

标题：数的平方等于两数乘积的方法数

出处：1577. 数的平方等于两数乘积的方法数

难度

5 级

题目描述

要求

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

• 类型 $\text{1}$：三元组 $\text{(i, j, k)}$，如果 ${\text{nums1[i]}}^{\text{2}}=\text{nums2[j]}\times \text{nums2[k]}$，其中 $\text{0}\le \text{i}<\text{nums1.length}$ 且 $\text{0}\le \text{j}<\text{k}<\text{nums2.length}$。
• 类型 $\text{2}$：三元组 $\text{(i, j, k)}$，如果 ${\text{nums2[i]}}^{\text{2}}=\text{nums1[j]}\times \text{nums1[k]}$，其中 $\text{0}\le \text{i}<\text{nums2.length}$ 且 $\text{0}\le \text{j}<\text{k}<\text{nums1.length}$。

示例

示例 1：

输入：$\text{nums1 = [7,4], nums2 = [5,2,8,9]}$
输出：$\text{1}$
解释：类型 $\text{1}$：$\text{(1,1,2)}$，${\text{nums1[1]}}^{\text{2}}=\text{nums2[1]}\times \text{nums2[2]}$（${\text{4}}^{\text{2}}=\text{2}\times \text{8}$）。

示例 2：

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

示例 3：

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

示例 4：

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

数据范围

• $\text{1}\le \text{nums1.length, nums2.length}\le \text{1000}$
• $\text{1}\le \text{nums1[i], nums2[i]}\le {\text{10}}^{\text{5}}$

解法

思路和算法

用 $(i,j,k)$ 表示符合题目要求的三元组。

如果 $(i,j,k)$ 是类型 $1$ 的三元组，则令 $x={\text{nums}}_1[i]$，$y={\text{nums}}_2[j]$，$z={\text{nums}}_2[k]$，有 $x^2=y\times z$。如果 $(i,j,k)$ 是类型 $2$ 的三元组，则令 $x={\text{nums}}_2[i]$，$y={\text{nums}}_1[j]$，$z={\text{nums}}_1[k]$，有 $x^2=y\times z$。

为了计算符合题目要求的三元组的数目，需要分别计算类型 $1$ 和类型 $2$ 的三元组的数目。假设数组 ${\text{nums}}_1$ 和 ${\text{nums}}_2$ 的长度分别是 $m$ 和 $n$。如果暴力统计符合题目要求的三元组，则统计类型 $1$ 的三元组需要 $O(m{n}^2)$ 的时间，统计类型 $2$ 的三元组需要 $O(n{m}^2)$ 的时间，总时间复杂度是 $O(mn(m+n))$，会超出时间限制，因此需要优化。

暴力方法的时间复杂度过高是因为相同的元素被重复计算。为了降低时间复杂度，应该避免重复计算。可以使用哈希表记录数组中的元素和每个元素的出现次数。

首先遍历数组 ${\text{nums}}_1$ 和 ${\text{nums}}_2$，分别用两个哈希表记录两个数组中每个元素的出现次数，用 ${\text{set}}_1$ 和 ${\text{set}}_2$ 分别表示两个数组的元素集合，然后计算两种类型的三元组的数目。两种类型的三元组的数目的计算方式大致相同。

计算类型 $1$ 的三元组的数目的方法如下：对于集合 ${\text{set}}_1$ 中的每个元素 $x$，遍历集合 ${\text{set}}_2$ 中的每个元素 $y$，如果集合 ${\text{set}}_2$ 中存在元素 $z$ 满足 $x^2=y\times z$ 且 $y$ 和 $z$ 在数组 ${\text{nums}}_2$ 中对应的下标不同，则 $x$、$y$ 和 $z$ 这三个元素组成一个类型 $1$ 的三元组。记 $x$ 在数组 ${\text{nums}}_1$ 中的出现次数为 $\text{xCount}$，$y$ 和 $z$ 在数组 ${\text{nums}}_2$ 中的出现次数分别为 $\text{yCount}$ 和 $\text{zCount}$，则可以根据出现次数计算该三元组的出现次数。为了避免重复计算，只考虑 $y\le z$ 的情况。该三元组的出现次数计算如下：

• 如果 $y=z$，则该三元组的出现次数是 $\text{xCount}\times \frac{\text{yCount}\times (\text{yCount}-1)}{2}$；

• 如果 $y<z$，则该三元组的出现次数是 $\text{xCount}\times \text{yCount}\times \text{zCount}$。

计算类型 $2$ 的三元组的数目的方法大致相同，区别是在集合 ${\text{set}}_2$ 中遍历元素 $x$，在集合 ${\text{set}}_1$ 中遍历元素 $y$ 并寻找元素 $z$。

得到类型 $1$ 和类型 $2$ 的三元组的数目之后，即可得到符合题目要求的三元组的数目。

代码

class Solution {
    public int numTriplets(int[] nums1, int[] nums2) {
        Map<Integer, Integer> map1 = new HashMap<Integer, Integer>();
        Map<Integer, Integer> map2 = new HashMap<Integer, Integer>();
        for (int num : nums1) {
            map1.put(num, map1.getOrDefault(num, 0) + 1);
        }
        for (int num : nums2) {
            map2.put(num, map2.getOrDefault(num, 0) + 1);
        }
        return numTriplets(map1, map2) + numTriplets(map2, map1);
    }

    public int numTriplets(Map<Integer, Integer> map1, Map<Integer, Integer> map2) {
        int triplets = 0;
        Set<Integer> set1 = map1.keySet();
        Set<Integer> set2 = map2.keySet();
        for (int x : set1) {
            long square = (long) x * x;
            int xCount = map1.get(x);
            for (int y : set2) {
                if (square % y == 0 && square / y <= Integer.MAX_VALUE) {
                    int z = (int) (square / y);
                    if (!map2.containsKey(z)) {
                        continue;
                    }
                    int yCount = map2.get(y);
                    if (y == z) {
                        triplets += xCount * (yCount * (yCount - 1) / 2);
                    } else if (y < z) {
                        int zCount = map2.get(z);
                        triplets += xCount * yCount * zCount;
                    }
                }
            }
        }
        return triplets;
    }
}

复杂度分析

• 时间复杂度：$O(mn)$，其中 $m$ 和 $n$ 分别是数组 ${\text{nums}}_1$ 和 ${\text{nums}}_2$ 的长度。遍历数组 ${\text{nums}}_1$ 和 ${\text{nums}}_2$ 并使用哈希表记录两个数组中的每个元素的出现次数需要 $O(m+n)$ 的时间，计算类型 $1$ 和类型 $2$ 的三元组的数目各需要 $O(mn)$ 的时间，因此总时间复杂度是 $O(mn)$。

• 空间复杂度：$O(m+n)$，其中 $m$ 和 $n$ 分别是数组 ${\text{nums}}_1$ 和 ${\text{nums}}_2$ 的长度。需要使用两个哈希表分别记录数组 ${\text{nums}}_1$ 和 ${\text{nums}}_2$ 中的每个元素的出现次数，两个哈希表的空间分别是 $O(m)$ 和 $O(n)$。