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;
}

优质内容筛选与推荐>>
1、MVC中关于Action过滤器小结
2、从一个表格文件中录入信息,进行计算后,在文件中输出这个表格
3、20155217 20155236 20155324 《信息安全系统设计基础》实验一 开发环境的熟悉
4、乱弹琴 Silverlight 2.0 (1) 开发环境准备
5、网站备忘


长按二维码向我转账

受苹果公司新规定影响,微信 iOS 版的赞赏功能被关闭,可通过二维码转账支持公众号。

    阅读
    好看
    已推荐到看一看
    你的朋友可以在“发现”-“看一看”看到你认为好看的文章。
    已取消,“好看”想法已同步删除
    已推荐到看一看 和朋友分享想法
    最多200字,当前共 发送

    已发送

    朋友将在看一看看到

    确定
    分享你的想法...
    取消

    分享想法到看一看

    确定
    最多200字,当前共

    发送中

    网络异常,请稍后重试

    微信扫一扫
    关注该公众号