【LeetCode】1994. 好子集的数目

题目

1994. 好子集的数目

给你一个整数数组 nums 。如果 nums 的一个子集中，所有元素的乘积可以表示为一个或多个 互不相同的质数 的乘积，那么我们称它为 好子集 。

• 比方说，如果

  nums = [1, 2, 3, 4]
  

  ：

  • [2, 3] ，[1, 2, 3] 和 [1, 3] 是 好 子集，乘积分别为 6 = 2*3 ，6 = 2*3 和 3 = 3 。
  • [1, 4] 和 [4] 不是 好 子集，因为乘积分别为 4 = 2*2 和 4 = 2*2 。

请你返回 nums 中不同的 好 子集的数目对 109 + 7 取余 的结果。

nums 中的 子集 是通过删除 nums 中一些（可能一个都不删除，也可能全部都删除）元素后剩余元素组成的数组。如果两个子集删除的下标不同，那么它们被视为不同的子集。

示例 1：

输入：nums = [1,2,3,4]
输出：6
解释：好子集为：
- [1,2]：乘积为 2 ，可以表示为质数 2 的乘积。
- [1,2,3]：乘积为 6 ，可以表示为互不相同的质数 2 和 3 的乘积。
- [1,3]：乘积为 3 ，可以表示为质数 3 的乘积。
- [2]：乘积为 2 ，可以表示为质数 2 的乘积。
- [2,3]：乘积为 6 ，可以表示为互不相同的质数 2 和 3 的乘积。
- [3]：乘积为 3 ，可以表示为质数 3 的乘积。

示例 2：

输入：nums = [4,2,3,15]
输出：5
解释：好子集为：
- [2]：乘积为 2 ，可以表示为质数 2 的乘积。
- [2,3]：乘积为 6 ，可以表示为互不相同质数 2 和 3 的乘积。
- [2,15]：乘积为 30 ，可以表示为互不相同质数 2，3 和 5 的乘积。
- [3]：乘积为 3 ，可以表示为质数 3 的乘积。
- [15]：乘积为 15 ，可以表示为互不相同质数 3 和 5 的乘积。

提示：

• 1 <= nums.length <= 105
• 1 <= nums[i] <= 30

思路

• 对于任何好子集，添加任意数目的1都是好子集

• 对于无平方数因子的数只能出现在好子集当中1次

• 任何有平方数因子的数都不能出现在好子集当中

• 设$dp[i][mask]$是第$i$个数，$mask$位质数被选择后的可能情况数，其状态转移方程为：
  $$
  dp[i][mask]=\left{\begin{aligned}

  &2^{freq(1)}, &i=1\
  &dp[i-1][mask], &num[i]\text{有平方因子} \
  &dp[i-1][mask] + dp[i-1][mask\oplus subset], &num[i]\text{无平方因子}
  \end{aligned}
  \right.
  $$

代码

from collections import Counter
from typing import List


class Solution:
    def numberOfGoodSubsets(self, nums: List[int]) -> int:
        primes = [2, 3, 5, 7, 11, 13, 17, 19, 23, 29]
        squares = [4, 8, 9, 12, 16, 18, 20, 24, 25, 27, 28]
        MOD = 10**9 + 7
        dp = [0] * (1 << 10)
        freqs = Counter(nums)
        dp[0] = pow(2, freqs[1], MOD)

        for n, freq in freqs.items():
            if n == 1 or n in squares:
                continue
            subset = 0
            for i, prime in enumerate(primes):
                if n % prime == 0:
                    subset |= (1 << i)
            for mask in range((1 << 10)-1, 0, -1):
                if (mask & subset) == subset:
                    dp[mask] = (dp[mask] + dp[mask ^ subset]*freq) % MOD
        return sum(dp[1:]) % MOD

复杂度

• 时间复杂度：$O(nC{2}^{\pi (C)})$ n为nums中非重复数的个数，C为$[2,\max (nums)]$质数的数量
• 空间复杂度：$O(2^{\pi (C)})$