Codeforces895 C. Square Subsets（状压dp）

最新推荐文章于 2023-09-11 16:52:55 发布

原创最新推荐文章于 2023-09-11 16:52:55 发布 · 259 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

本文探讨了一种使用状压动态规划解决特定数学问题的方法，即计算给定序列中积为完全平方数的子集数量。通过分析数据范围和质数特性，提出了一种高效的算法解决方案。

题意：

给定长度为n的序列，
问有多少个子集，满足子集的积是完全平方数，
答案对1e9+7取模

数据范围：n<=1e5，1<=a(i)<=70

解法：

观察到a(i)<=70，打表发现70以内的质数只有19个,容易想到状压.

完全平方数的性质是每个质因子的指数都是偶数,
那么如果用19位二进制数表示质因子的奇偶性,那么完全平方数就是全0,
那么这题可以搞一波状压dp.

令d[i][j]表示前i个数,子集乘积状态为j的方案数.
然而第一维1e5,第二维1e6,空间都开不下.

再观察数据范围:a(i)<=70,
同时想到对于一个数x,x的质因子奇偶性和x^3一样,x^2的质因子奇偶性和x^4一样.
也就是说只需要考虑奇数次幂和偶数次幂的情况.
那么可以用大小为70的桶存每个数的数量
那么第一维可以降到70.

对于当前数x,设数量为cnt,我们只需要考虑选出奇数个还是偶数个:
1.如果选择奇数个,方案数为2^(cnt-1)
2.如果选择偶数个,方案数为2^(cnt-1)

最后就是普通状压dp了,
令d[i][j]表示前i个数,子集乘积传统为j的方案数,

最后答案是d[x][0],其中x为用到的数的数量,
这题直接把第一维滚动掉更好写.

最后记得减去空集,空集不算

code：

#include<bits/stdc++.h>
using namespace std;
#define int long long
const int maxm=1e5+5;
const int mod=1e9+7;
int prime[maxm],num;
int d[2][1<<20];
int p[maxm];
int cnt[77];
int a[maxm];
int n;
int isprime(int x){
    for(int i=2;i*i<=x;i++){
        if(x%i==0)return 0;
    }
    return 1;
}
void init(){
    for(int i=2;i<=70;i++){
        if(isprime(i)){
            prime[num++]=i;
        }
    }
    //预处理pow2(i)
    p[0]=1;
    for(int i=1;i<maxm;i++){
        p[i]=p[i-1]*2%mod;
    }
}
int ppow(int a,int b,int mod){
    int ans=1%mod;a%=mod;
    while(b){
        if(b&1)ans=ans*a%mod;
        a=a*a%mod;
        b>>=1;
    }
    return ans;
}
int cal(int x){
    int ans=0;
    for(int i=0;i<num;i++){
        if(x%prime[i]==0){
            while(x%prime[i]==0){
                ans^=(1<<i);
                x/=prime[i];
            }
        }
    }
    return ans;
}
signed main(){
    init();
    cin>>n;
    for(int i=1;i<=n;i++){
        cin>>a[i];
        cnt[a[i]]++;
    }
    int x=0,y=1;
    d[x][0]=1;
    for(int i=1;i<=70;i++){
        if(!cnt[i])continue;
        int v=cal(i);
        for(int j=0;j<(1<<num);j++){//选奇数个还是偶数个都是2^(cnt[i]-1)
            d[y][j]=p[cnt[i]-1]*(d[x][j]+d[x][j^v])%mod;
        }
        swap(x,y);
    }
    int ans=(d[x][0]-1+mod)%mod;//减掉空集
    cout<<ans<<endl;
    return 0;
}