力扣 1994. 好子集的数目数学状压dp

最新推荐文章于 2022-11-12 10:38:41 发布

csu_xiji

最新推荐文章于 2022-11-12 10:38:41 发布

阅读量262

点赞数

分类专栏：力扣状压dp 数论文章标签：动态规划算法

本文链接：https://blog.csdn.net/xiji333/article/details/120170158

版权

https://leetcode-cn.com/problems/the-number-of-good-subsets/
在这里插入图片描述
思路一：纯数学暴力解法。首先我们知道一个合数一定可以表示为若干个素数的乘积，因为要求素数不能相同，因此子集中的合数一定不能表示成两个相同素数的乘积。注意到元素的取值范围为 $[1, 30]$ ，据此我们可以排除一些合数。然后把相同的数合并到一起(记录出现的次数)，那么就只剩下大概20个元素了。此时可以暴力枚举选择的情况，然后对选择的数做因数分解统计每个质因子出现的次数，如果不合法就不计数。注意如果合法的话，由于每个元素可能出现了多次，任意选择某一个都是一种新的情况，因此计数时要使用乘法原理。最后特殊考虑数字1，因为任意一种合法情况加上任意个1都是合法的，所以我们可以把1放到最后考虑，不妨设1出现的次数为 $c n t$ ，其他情况的计数总和为 $s u m$ ，最终答案就等于 $sum*2^{cnt}$ 。总复杂度为 $O(n*2^{n})$ ，其中 $n$ 为合法的数字数量。

class Solution {
   
public:
    const int mod=1e9+7;
    using ll=long long;
    int qpow(ll a,ll b)
    {
   
        ll ans=1;
        while(b)
        {
   
            if(b&1)
                ans=(ans*a)%mod;
            b>>=1;
            a=(a*a)%mod;
        }
        return ans;
    }
    
    int numberOfGoodSubsets(vector<int>& vec) {
   
        set<int> target={
   2,3,5,6,7,10,11,13,14,15,17,19,21,22,23,26,29,30};
        vector<int> cf[31];
        cf[2]={
   2};
        cf[3]={
   3};
        cf[5]={
   5};
        cf[6]={
   2,3};
        cf[7]={
   7};
        cf[10]={
   2,5};
        cf[11]={
   11};
        cf[13]={
   13};
        cf[14]={
   2,7};
        cf[15]={
   3,5};
        cf[17]