BZOJ4652 NOI2016循环之美(莫比乌斯反演+杜教筛)

  因为要求数值不同,不妨设gcd(x,y)=1。由提示可以知道,x/y是纯循环小数的充要条件是x·klen=x(mod y)。因为x和y互质,两边同除x,得klen=1(mod y)。那么当且仅当k和y互质,存在len使该式成立。

  于是现在要求的就是

  k是固定的,先不管后面一部分。套路地化式子:

  

  

  

  设f(i)=[i⊥k]。注意到k很小,并且显然有gcd(j,k)=gcd(j%k,k)。于是O(k)的预处理出f的前缀和。

  

  那么几乎已经做到线性了,能拿到84分,感觉非常棒。

  然而要A掉还需要低于线性的做法。看到两个下取整就会特别想整除分块。那么现在我们需要求的是g(i)=μ(i)*[i⊥k]的前缀和。

  继续套路地化:

  

  

  

  注意到若i和d不互质,则μ(id)=0,否则μ(i)·μ(d)=μ(id)。

  

  可以发现后面一个求和和我们的g函数形式上是一样的。于是可以递归求解,记忆化一下。边界k=1时用杜教筛求出μ的前缀和。

  考虑复杂度。上面式子中的n有√N种(⌊N/d⌋的取值个数),k有√K种(K的因子个数)。转移时枚举因子√K种。把μ筛到2e7的话,杜教筛的次数非常少。于是总复杂度约为O(k√n+n2/3),且远远跑不满。

  以上复杂度分析均为口胡。总之O(能过)。

 

#include<iostream> 
#include<cstdio>
#include<cmath>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<map>
using namespace std;
int read()
{
    int x=0,f=1;char c=getchar();
    while (c<'0'||c>'9') {if (c=='-') f=-1;c=getchar();}
    while (c>='0'&&c<='9') x=(x<<1)+(x<<3)+(c^48),c=getchar();
    return x*f;
}
#define N 20000010
#define K 2010
int n,m,k,mobius[N],prime[N],sum[K],cnt=0;
long long ans=0;
bool flag[N];
map<int,int> miu;
map<int,int> G[K];
int gcd(int n,int m){return m==0?n:gcd(m,n%m);}
inline int f(int n){return sum[k]*(n/k)+sum[n%k];}
int calc(int k)
{
    if (k<=min(min(n,m),N-10)) return mobius[k];
    if (miu[k]) return miu[k];
    int s=1;
    for (int i=2;i<=k;i++)
    {
        int t=k/(k/i);
        s-=(t-i+1)*calc(k/i);
        i=t;
    }
    miu[k]=s;
    return s;
}
int g(int n,int k)
{
    if (n==0) return 0;
    if (G[k][n]) return G[k][n];
    if (k==1) return calc(n);
    int s=0;
    for (int i=1;i*i<=k;i++)
    if (k%i==0)
    {
        if (mobius[i]-mobius[i-1]) s+=g(n/i,i);
        if (i*i!=k&&(mobius[k/i]-mobius[k/i-1])) s+=g(n/(k/i),k/i);
    }
    G[k][n]=s;
    return s;
}
int main()
{
#ifndef ONLINE_JUDGE
    freopen("bzoj4652.in","r",stdin);
    freopen("bzoj4652.out","w",stdout);
    const char LL[]="%I64d\n";
#else
    const char LL[]="%lld\n";
#endif
    n=read(),m=read(),k=read();
    flag[1]=1;mobius[1]=1;
    for (int i=2;i<=max(min(min(n,m),N-10),k);i++)
    {
        if (!flag[i]) prime[++cnt]=i,mobius[i]=-1;
        for (int j=1;j<=cnt&&prime[j]*i<=min(min(n,m),N-10);j++)
        {
            flag[prime[j]*i]=1;
            if (i%prime[j]==0) break;
            else mobius[prime[j]*i]=-mobius[i];
        }
    }
    for (int i=2;i<=max(min(min(n,m),N-10),k);i++) mobius[i]+=mobius[i-1];
    for (int i=1;i<=k;i++)
    sum[i]=sum[i-1]+(gcd(i,k)==1);
    for (int i=1;i<=min(n,m);i++)
    {
        int t=min(n/(n/i),m/(m/i));
        ans+=1ll*(n/i)*f(m/i)*(g(t,k)-g(i-1,k));
        i=t;
    }
    cout<<ans;
    return 0;
}

 

posted @ 2018-08-16 17:00  Gloid  阅读(128)  评论(0编辑  收藏  举报