BZOJ5019 SNOI2017遗失的答案（容斥原理）

　　显然存在方案的数一定是L的因数，考虑对其因子预处理答案，O(1)回答。

　　考虑每个质因子，设其在g中有x个，l中有y个，则要求所有选中的数该质因子个数都在[x,y]中，且存在数的质因子个数为x、y。对于后一个限制，显然可以简单地容斥，即[x,y]-[x+1,y]-[x,y-1]+[x+1,y-1]，枚举这个至多是4⁸的，这个取最大值时因子个数是2⁸。暴力枚举数数即可。复杂度总之O(能过)。

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<cstdlib>
#include<algorithm>
using namespace std;
#define P 1000000007
#define ll long long
int read()
{
    int x=,f=;char c=getchar();
    while (c<''||c>'') {if (c=='-') f=-;c=getchar();}
    while (c>=''&&c<='') x=(x<<)+(x<<)+(c^),c=getchar();
    return x*f;
}
int n,g,l,u,m,prime[],cnt[][],d[],ans[],p[][],q[][],a[],tot,t,sum;
int ksm(int a,int k)
{
    int s=;
    for (;k;k>>=,a=1ll*a*a%P) if (k&) s=1ll*s*a%P;
    return s;
}
void get(int k,int s)
{
    if (k>t) {sum++;return;}
    ll x=ksm(prime[k],cnt[][k]);
    for (int i=cnt[][k];i<=cnt[][k];i++)
    {
        if (s*x<=n) get(k+,s*x);else break;
        x=1ll*x*prime[k];
    }
}
void build(int k,int s)
{
    if (k>t)
    {
        p[++tot][]=s;
        for (int i=;i<=t;i++) p[tot][i]=a[i];
        return;
    }
    ll x=ksm(prime[k],cnt[][k]);
    for (int i=cnt[][k];i<=cnt[][k];i++)
    {
        a[k]=i;
        if (s*x<=n) build(k+,s*x);else break;
        x=1ll*x*prime[k];
    }
}
void calc(int op)
{
    //for (int i=1;i<=t;i++) cout<<prime[i]<<' '<<cnt[0][i]<<' '<<cnt[1][i]<<endl;cout<<endl;
    sum=;get(,);
    for (int i=;i<=tot;i++)
    {
        bool flag=;
        for (int j=;j<=t;j++)
        if (p[i][j]<cnt[][j]||p[i][j]>cnt[][j]) {flag=;break;}
        if (flag)
        {
            ans[i]+=op*ksm(,sum-);
            if (ans[i]<) ans[i]+=P;if (ans[i]>=P) ans[i]-=P;
        }
    }
}
void dfs(int k,int op)
{
    if (k>t) {calc(op);return;}
    dfs(k+,op);
    cnt[][k]++;dfs(k+,-op);
    cnt[][k]--;dfs(k+,op);
    cnt[][k]--;dfs(k+,-op);
    cnt[][k]++;
}
int main()
{
#ifndef ONLINE_JUDGE
    freopen("bzoj5019.in","r",stdin);
    freopen("bzoj5019.out","w",stdout);
    const char LL[]="%I64d\n";
#else
    const char LL[]="%lld\n";
#endif
    n=read(),g=read(),l=read(),m=read();
    if (l%g) {for (int i=;i<=m;i++) printf("0\n");return ;}
    u=l;
    for (int i=;i*i<=u;i++)
    if (u%i==)
    {
        prime[++t]=i,cnt[][t]=;u/=i;
        while (u%i==) cnt[][t]++,u/=i;
    }
    if (u>) prime[++t]=u,cnt[][t]=;
    u=g;
    for (int i=;i<=t;i++)
    while (u%prime[i]==) u/=prime[i],cnt[][i]++;
    build(,);
    for (int i=;i<=tot;i++) d[i]=p[i][];
    sort(d+,d+tot+);
    for (int i=;i<=tot;i++)
        for (int j=;j<=t;j++)
        q[i][j]=p[i][j];
    for (int i=;i<=tot;i++)
    {
        int x=lower_bound(d+,d+tot+,q[i][])-d;
        for (int j=;j<=t;j++) p[x][j]=q[i][j];
    }
    dfs(,);
    while (m--)
    {
        int x=read(),y=lower_bound(d+,d+tot+,x)-d;
        if (d[y]!=x) {printf("0\n");continue;}
        else printf("%d\n",ans[y]);
    }
    return ;
}