luogu P5366 [SNOI2017]遗失的答案

luogu

首先gcd为\(G\),lcm为\(L\),有可能出现的数可以发现只有几百个.如果选出的数要能取到gcd,那么对于每种质因子,都要有一个数的这种质因子出现次数取到最小值,要取到lcm,也就是每种质因子都有数达到其出现次数的最大值.那我们给每个合法的数一个二进制权值,表示这个数的某种质因子出现次数是否为这种质因子出现次数最小值/最大值

那么现在问题就是选出若干个数,使得权值或起来为全集的方案,设\(f_i\)为或起来为\(i\)的答案.直接做复杂度有点大,这里再设\(g_i\)为或起来为\(i\)的子集的答案,因为权值二进制为\(i\)子集的数都可以选或不选,所以如果\(cnt_i\)为权值二进制为\(i\)子集的数个数,那么\(g_i=2^{cnt_i}\).\(cnt\)可以高维前缀和求得.然后要由\(g\)推到\(f\),因为\(g\)恰好是\(f\)的高维前缀和形式,所以可以容斥求得\(f\),即\(f_s=\sum_{t\subseteq s} (-1)^{|s|-|t|}g_t\)

现在是每次强制要选一个数,使得子集权值或起来为全集.那么因为选了这个数,那我们只要考虑数\(i\)对应权值为0的位就好了.如果记全集为\(U\),答案大概可以写成这样\(ans_s=\sum_{t} (-1)^{|U-s|-|t-(t\cap s)|}{g'}_t\),其中\({g'}_t\)为不包含给定的数,或起来为\(t\)的子集的答案,这个就是把包含给定数权值\(s\)\(g_t\)要除以2,剩下的不变得到

#include<bits/stdc++.h>
#define LL long long
#define uLL unsigned long long
#define db double

using namespace std;
const int N=1000+10,M=(1<<16)+10,mod=1e9+7,inv2=500000004;
int rd()
{
    int x=0,w=1;char ch=0;
    while(ch<'0'||ch>'9') {if(ch=='-') w=-1;ch=getchar();}
    while(ch>='0'&&ch<='9') {x=(x<<3)+(x<<1)+(ch^48);ch=getchar();}
    return x*w;
}
int fpow(int a,int b){int an=1;while(b){if(b&1) an=1ll*an*a%mod;a=1ll*a*a%mod,b>>=1;} return an;}
map<int,int> id;
int n,m,gg,ll,pg[10][2],pl[10][2],tt;
int f[N],an[N],g[M],pc[M];
void inii(int o,int s,int zt)
{
    if(o>tt){if(s<=n) id[s]=++m,f[m]=zt,++g[zt];return;}
    int j=1;
    for(int i=1;i<=pg[o][1];++i) j*=pg[o][0];
    for(int i=pg[o][1];i<=pl[o][1];++i,j*=pg[o][0])
        inii(o+1,s*j,zt|((i==pg[o][1])<<(o-1))|((i==pl[o][1])<<(o+tt-1)));
}

int main()
{
    n=rd(),gg=rd(),ll=rd();
    int x=ll,sqt=sqrt(x);
    for(int i=2;i<=sqt;++i)
        if(x%i==0)
        {
            pl[++tt][0]=i;
            while(x%i==0) ++pl[tt][1],x/=i;
            if(x==1) break;
        }
    if(x>1) pl[++tt][0]=x,pl[tt][1]=x;
    x=gg;
    for(int i=1;i<=tt;++i)
    {
        pg[i][0]=pl[i][0];
        while(x%pg[i][0]==0) ++pg[i][1],x/=pg[i][0];
    }
    inii(1,1,0);
    int nn=1<<(tt<<1),u=nn-1;
    for(int j=1;j<nn;j<<=1)
        for(int i=0;i<nn;++i)
            if(i&j) g[i]=g[i]+g[i^j];
    for(int i=0;i<nn;++i) g[i]=fpow(2,g[i]);
    for(int i=1;i<nn;++i) pc[i]=pc[i^(i&(-i))]+1;
    for(int i=1;i<=m;++i)
    {
        int zt=f[i]^u,tc=pc[zt];
        for(int j=u;j;--j)
            an[i]=(an[i]+1ll*g[j]*(inv2*(bool)((j&(zt^u))==(zt^u)))%mod*((tc-pc[j&zt])&1?-1:1)+mod)%mod;
    }
    int q=rd();
    while(q--)
    {
        int x=rd();
        if(!id.count(x)) puts("0");
        else printf("%d\n",an[id[x]]);
    }
    return 0;
}

猜你喜欢

转载自www.cnblogs.com/smyjr/p/11567781.html