codeforces895C Square Subsets 状压dp

题目链接:戳这里

题目大意:Petya又迟到了...老师给了他额外的任务。对于一个数组a,Petya需要统计从中间选择非空子集,使它们的乘积等于某个整数的平方的方法的数量。 如果这些方法所选择的元素的索引不同,则认为这两种是不同的方法。 因为结果可能很大,结果需要mod 10^9+7。

题解:一个数是完全平方数,说明它的每个质因子都有偶数个,而a数组中的每个元素最大不超过70,那么我们可以对1~70分解质因数,用2^19来表示有奇数还是偶数个质因子,至于为什么是2^19是因为70以内有19个质数。然后就可以做状压dp转移,我们可以统计每个数出现的次数,这样就相当于是缩小了n的范围(70以内),这样数组才能开的下qwq。

用dp[i][state]表示考虑到i,当前构成的数的状态是state的方案数。

如果i这个数没有出现过,那么直接和上一步转移过来的值相等。

如果出现过,那么考虑选和不选,显然都有上一步*2^(出现次数-1)种方案,分别统计即可。

最后答案是dp[70][0]-1,因为0不算在内。

代码:

#include<bits/stdc++.h>
#define mod 1000000007
using namespace std;
typedef long long LL;
int read()
{
    char c;int sum=0,f=1;c=getchar();
    while(c<'0' || c>'9'){if(c=='-')f=-1;c=getchar();}
    while(c>='0' && c<='9'){sum=sum*10+c-'0';c=getchar();}
    return sum*f;
}
int prime[19]={2,3,5,7,11,13,17,19,23,29,31,37,41,43,47,53,59,61,67};
int n,x,cnt[75];
int dp[75][(1<<19)],bin[100005];
int s[75]; 
void init()
{
    for(int i=1;i<=70;i++)
    {
        for(int j=0;j<19;j++)
        {
        	int t=i;
            while(t%prime[j]==0)
            t/=prime[j],s[i]^=(1<<j);
        }
    }
    bin[0]=1;
    for(int i=1;i<=n;i++)
    bin[i]=(bin[i-1]*2)%mod;
}
int main()
{
    n=read();
    for(int i=1;i<=n;i++)
    {
        x=read();
        cnt[x]++;
    }
    init();
    dp[0][0]=1;
    for(int i=1;i<=70;i++)
    {
        if(!cnt[i])
        {
            for(int j=0;j<(1<<19);j++)
            dp[i][j]=dp[i-1][j];
        }
        else
        {
            for(int j=0;j<(1<<19);j++)
            {
                dp[i][j^s[i]]=(dp[i][j^s[i]]+(LL)bin[cnt[i]-1]*dp[i-1][j])%mod;
                dp[i][j]=(dp[i][j]+(LL)bin[cnt[i]-1]*dp[i-1][j])%mod;
            }
        }
    }
    cout<<(dp[70][0]-1)%mod<<endl;
    return 0;
}

猜你喜欢

转载自blog.csdn.net/qq_39791208/article/details/79383936