欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

HDU 6439 2018CCPC网络赛 Congruence equationI(杜教筛 + 莫比乌斯反演 + 伯努利数)

程序员文章站 2024-02-22 12:16:29
...

HDU 6439 2018CCPC网络赛 Congruence equationI(杜教筛 + 莫比乌斯反演 + 伯努利数)

 

 

大致题意:给你一个长度为k的序列a。对于序列c,当 HDU 6439 2018CCPC网络赛 Congruence equationI(杜教筛 + 莫比乌斯反演 + 伯努利数) 时,HDU 6439 2018CCPC网络赛 Congruence equationI(杜教筛 + 莫比乌斯反演 + 伯努利数);当HDU 6439 2018CCPC网络赛 Congruence equationI(杜教筛 + 莫比乌斯反演 + 伯努利数)时,HDU 6439 2018CCPC网络赛 Congruence equationI(杜教筛 + 莫比乌斯反演 + 伯努利数)取[0,m)中任意一个数字。令 HDU 6439 2018CCPC网络赛 Congruence equationI(杜教筛 + 莫比乌斯反演 + 伯努利数) 表示满足 HDU 6439 2018CCPC网络赛 Congruence equationI(杜教筛 + 莫比乌斯反演 + 伯努利数) 的序列c的方案数。现在让你求 HDU 6439 2018CCPC网络赛 Congruence equationI(杜教筛 + 莫比乌斯反演 + 伯努利数)

                                                          


 

首先,根据裴蜀定理,满足HDU 6439 2018CCPC网络赛 Congruence equationI(杜教筛 + 莫比乌斯反演 + 伯努利数)的条件是HDU 6439 2018CCPC网络赛 Congruence equationI(杜教筛 + 莫比乌斯反演 + 伯努利数),那么我们不妨分为两种情况处理。对于HDU 6439 2018CCPC网络赛 Congruence equationI(杜教筛 + 莫比乌斯反演 + 伯努利数)的数字,假设他们的gcd为g,那么剩下的数字与g的gcd就要是1。设HDU 6439 2018CCPC网络赛 Congruence equationI(杜教筛 + 莫比乌斯反演 + 伯努利数)的项有k个,加上这个m,设这k+1个数字的gcd为d,那么gcd(d,g)要等于1。由于这k+1个数字里面有一个定值m,所以这个d一定是m的因子。我们令f(d)表示这k+1个数字的gcd为d的方案数。那么开始第一次莫比乌斯反演,有:

                                                         HDU 6439 2018CCPC网络赛 Congruence equationI(杜教筛 + 莫比乌斯反演 + 伯努利数)

HDU 6439 2018CCPC网络赛 Congruence equationI(杜教筛 + 莫比乌斯反演 + 伯努利数),那么最后的答案就是:

                                                           HDU 6439 2018CCPC网络赛 Congruence equationI(杜教筛 + 莫比乌斯反演 + 伯努利数)

HDU 6439 2018CCPC网络赛 Congruence equationI(杜教筛 + 莫比乌斯反演 + 伯努利数),于是我们开始第二次莫比乌斯反演:

                                                  HDU 6439 2018CCPC网络赛 Congruence equationI(杜教筛 + 莫比乌斯反演 + 伯努利数)

这样,对于g,我们只需要用到gcd为1的,所以我们不妨把第二个参数去掉。整理一下,最后的答案就是:

                                                          HDU 6439 2018CCPC网络赛 Congruence equationI(杜教筛 + 莫比乌斯反演 + 伯努利数)

对于这个g(x),根据莫比乌斯函数的性质,有效的i肯定不会是任意一个质因子的2次幂及以上,所以i一定是x的质因子的线性组合,因此我们可以把这个预处理出来。然后 HDU 6439 2018CCPC网络赛 Congruence equationI(杜教筛 + 莫比乌斯反演 + 伯努利数) 最多只有HDU 6439 2018CCPC网络赛 Congruence equationI(杜教筛 + 莫比乌斯反演 + 伯努利数)个,可以分块求和。假设 HDU 6439 2018CCPC网络赛 Congruence equationI(杜教筛 + 莫比乌斯反演 + 伯努利数) 的质因子数为cnt,那么计算g部分的开销就是HDU 6439 2018CCPC网络赛 Congruence equationI(杜教筛 + 莫比乌斯反演 + 伯努利数),本题cnt最大为9,完全可以接受。那么问题的关键就是如何求h(d)的前缀和了。

我们发现,如果令HDU 6439 2018CCPC网络赛 Congruence equationI(杜教筛 + 莫比乌斯反演 + 伯努利数),那么h(d)就是H(d)和HDU 6439 2018CCPC网络赛 Congruence equationI(杜教筛 + 莫比乌斯反演 + 伯努利数)的迪利克雷卷积,而H(d)和HDU 6439 2018CCPC网络赛 Congruence equationI(杜教筛 + 莫比乌斯反演 + 伯努利数)是显然具有积性的。所以说我们可以用积性函数的性质,构造线性筛来求解h(d)的前缀和。但是注意到,本题的数据范围是HDU 6439 2018CCPC网络赛 Congruence equationI(杜教筛 + 莫比乌斯反演 + 伯努利数),而且还有多组数据,即使是线性的筛法也无法满足条件。所以我们这里考虑用杜教筛。

杜教筛了解一下:https://blog.csdn.net/skywalkert/article/details/50500009

简单来说就是,利用迪利克雷卷积的恒等变换,使得一个原本不容易求的积性数论函数的前缀和,变成两个容易求的积性数论函数运算,最后转化成HDU 6439 2018CCPC网络赛 Congruence equationI(杜教筛 + 莫比乌斯反演 + 伯努利数)的形式,其中A(n)表示辅助的容易求和的积性数论函数的前缀和。之后对于前HDU 6439 2018CCPC网络赛 Congruence equationI(杜教筛 + 莫比乌斯反演 + 伯努利数)的S(n),用构造线性筛直接打表计算;对于后面的大的部分,利用上面的式子分块记忆化搜索计算。可以证明这样子做的复杂度是O(HDU 6439 2018CCPC网络赛 Congruence equationI(杜教筛 + 莫比乌斯反演 + 伯努利数))的。

对于本题,令HDU 6439 2018CCPC网络赛 Congruence equationI(杜教筛 + 莫比乌斯反演 + 伯努利数),我们可以这么推导:

                                HDU 6439 2018CCPC网络赛 Congruence equationI(杜教筛 + 莫比乌斯反演 + 伯努利数)

最后这个东西满足杜教筛的形式,关于前面这个幂和,我们同样有方法计算:https://www.zybuluo.com/yang12138/note/848419

本题的话选用伯努利数的方法来计算幂和,即 HDU 6439 2018CCPC网络赛 Congruence equationI(杜教筛 + 莫比乌斯反演 + 伯努利数)。经过曲折之后,我们就可以在O(HDU 6439 2018CCPC网络赛 Congruence equationI(杜教筛 + 莫比乌斯反演 + 伯努利数))的时间复杂度内求出这个h(d)的前缀和S(d)。最后的答案HDU 6439 2018CCPC网络赛 Congruence equationI(杜教筛 + 莫比乌斯反演 + 伯努利数),分成两部分分块求和即可。最后总的时间复杂度是HDU 6439 2018CCPC网络赛 Congruence equationI(杜教筛 + 莫比乌斯反演 + 伯努利数)。具体见代码:

#include<bits/stdc++.h>
#define LL long long
#define mod 1000000007
#define pb push_back
#define lb lower_bound
#define ub upper_bound
#define INF 0x3f3f3f3f
#define sf(x) scanf("%d",&x)
#include <ext/pb_ds/hash_policy.hpp>
#include <ext/pb_ds/assoc_container.hpp>
#define sc(x,y,z) scanf("%d%d%d",&x,&y,&z)
#define clr(x,n) memset(x,0,sizeof(x[0])*(n+5))
#define file(x) freopen(#x".in","r",stdin),freopen(#x".out","w",stdout)

using namespace __gnu_pbds;
using namespace std;

const int N = 1e6 + 10;
const int K = 1e2 + 10;

int n,m,k,tot,S[N],C[K][K],p[N];
gp_hash_table<int,int> mp;
std::vector<int> pri_fac;
typedef pair<int,int> P;
LL B[K],invk,pw[N];
std::vector<P> fac;
bool isp[N];

LL qpow(LL x,int n)
{
    LL res=1;
    while(n)
    {
        if(n&1) res=res*x%mod;
        x=x*x%mod; n>>=1;
    }
    return res;
}

void init()
{
    C[0][0]=1;
    for(int i=1;i<K;i++) C[i][0]=1;
    for(int i=1;i<K;i++)
        for(int j=1;j<K;j++)
            C[i][j]=(C[i-1][j]+C[i-1][j-1])%mod;
    B[0]=1;
    for(int i=1;i<K;i++)
    {
        for(int j=0;j<i;j++)
            B[i]+=C[i+1][j]*B[j]%mod;
        B[i]%=mod; B[i]=(mod-B[i])*qpow(i+1,mod-2)%mod;
    }
}

void sieve(int n)
{
    int sz=0; S[1]=1;
    for(int i=2;i<n;i++)
    {
        if (!isp[i])
        {
            p[++sz]=i;
            pw[i]=qpow(i,k);
            S[i]=pw[i]-1;
        }
        for(int j=1;j<=sz&&p[j]*i<n;j++)
        {
            int x=i*p[j]; isp[x]=1;
            if (i%p[j]==0)
            {
                S[x]=S[i]*pw[p[j]]%mod;
                break;
            } else S[x]=S[i]*(pw[p[j]]-1)%mod;
        }
    }
    for(int i=2;i<n;i++) S[i]=(S[i-1]+S[i])%mod;
}

int powsum(int x)
{
    if (k==0) return x;
    LL res=0,pw=1;
    for(int i=1;i<=k+1;i++)
    {
        pw=pw*(x+1)%mod;
        res+=C[k+1][i]*B[k+1-i]%mod*pw%mod;
    }
    res%=mod;
    return res*invk%mod;
}

int s(int x)
{
    if (x<tot) return S[x];
    if (mp[x]) return mp[x];
    LL res=powsum(x);
    for(int l=2,r;l<=x;l=r+1)
    {
        r=x/(x/l);
        res-=(r-l+1)*(LL)s(x/l)%mod;
    }
    res=res%mod+mod;
    return mp[x]=res%mod;
}

LL cal(int x)
{
    LL res=0;
    for(auto i:fac)
    {
        if (i.first>x) break;
        res+=(x/i.first)*(i.second&1?-1:1);
    }
    return res%mod;
}

int main()
{
    int T; sf(T);
    init(); int AC=1;
    while(T--)
    {
        sf(n); sf(m);
        int g=0; k=0; mp.clear();
        for(int i=1;i<=n;i++)
        {
            int x; sf(x);
            if(x==-1)k++;else g=__gcd(g,x);
        }
        sieve(tot=ceil(pow(m,2.0/3)));
        invk=qpow(k+1,mod-2);
        if (g==0)
        {
            printf("Case #%d: %d\n",AC++,s(m));
            continue;
        }
        LL ans=0;
        fac.clear();
        pri_fac.clear();
        for(int i=2;i*i<=g;i++)
        {
            if (g%i) continue;
            pri_fac.pb(i);while(g%i==0) g/=i;
        }
        if (g>1) pri_fac.pb(g);
        int up=1<<pri_fac.size();
        for(int i=0;i<up;i++)
        {
            int cnt=0,d=1;
            for(int j=0;j<pri_fac.size();j++)
                if (i&(1<<j)) cnt++,d*=pri_fac[j];
            fac.pb(P(d,cnt));
        }
        sort(fac.begin(),fac.end());
        int pre=0,cur;
        for(int l=1,r;l<=m;l=r+1)
        {
            r=m/(m/l); cur=s(r);
            ans+=(cur-pre)*cal(m/l)%mod;
            pre=cur;
        }
        ans%=mod;
        printf("Case #%d: %lld\n",AC++,(ans+mod)%mod);
    }
    return 0;
}