徐州网络赛 D.Easy Math (莫比乌斯反演+杜教筛)*

题目链接：https://nanti.jisuanke.com/t/31456

#include<bits/stdc++.h>
#pragma comment(linker,"/STACK:1024000000,1024000000")
using namespace std;

#define debug puts("YES");
#define rep(x,y,z) for(int (x)=(y);(x)<(z);(x)++)
#define read(x,y) scanf("%d%d",&x,&y)

#define lrt int l,int r,int rt
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
#define ll long long
const int  maxn =1e7+5;
const int mod=1e9+7;
ll powmod(ll x,ll y) {ll t;for(t=1;y;y>>=1,x=x*x%mod) if(y&1) t=t*x%mod;return t;}
ll gcd(ll x,ll y)  {  return y==0?x:gcd(y,x%y); }
/*
题目大意：简单的数学求和意思。

这道题考的知识面挺广的，
首先要学会式子的推算，
原式=sigma miu[i]*miu[n]*[gcd(i,n)==1]。
后面最典型的反演特征出来了，
一系列操作后不难得到，
f(n,m)=miu[n]*sigma miu[d]*f(d,m/d)。
这个结构可以递归求解，
最终子状态是f(1,m)=梅（xx）函数，就是莫比乌斯函数前缀和
f(n,1)就是miu[n]。

这样时间复杂度看起来蛮可怕的，，
但仔细推敲下，
f函数中的n要求是无平方因子数，
范围达到九次方，那么可以知道f递归最多展开九层不到。

下面就是杜教筛的套路了，用map结构记忆化，
并且预处理7次方的和形式，还对n进行分块处理。

还有个东西就是求mobius函数，直接根号复杂度枚举质因子即可。
*/
///筛法筛莫比乌斯函数及其前缀和
int prim[maxn],tot=0;
int vis[maxn],miu[maxn],smu[maxn];
void sieve()
{
    miu[1]=1;
    for(int i=2;i<maxn;i++)
    {
        if(vis[i]==0) prim[tot++]=i,miu[i]=-1;
        for(int j=0;j<tot;j++)
        {
            if(1LL*i*prim[j]>=maxn) break;
            int k=i*prim[j];
            vis[k]=-1;
            if(i%prim[j]) miu[k]=-miu[i];
            else break;
        }
    }
    for(int i=1;i<maxn;i++) smu[i]=smu[i-1]+miu[i];
}
///杜教筛记忆化搜索
map<ll,ll> dp;///记忆化结构
ll ms(ll x)///记忆化计算miu函数的前缀和
{
    if(x<maxn) return dp[x]=smu[x];///预处理好的结构
    if(dp[x]) return dp[x];
    ll ans=1;
    for(ll i=2,j;i<=x;i=j+1)
    {
        j=x/(x/i);
        ans-=(j-i+1)*ms(x/i);
    }
    return dp[x]=ans;
}
///答案函数主体
int getm(ll x)
{
    int ret=1;
    if(x<maxn) return miu[x];
    for(ll i=2;1LL*i*i<=x;i++)
        if(x%i==0)
        {
            if(x%(i*i)==0) return 0;
            x/=i,ret*=-1;
        }
    if(x>1) ret*=-1;
    return ret;
}
ll f(ll x,ll y)///n,m
{
    if(x==1) return ms(y);
    if(y==1) return getm(x);///递归子状态

    ll tm=getm(x),ans=0;
    if(tm==0) return tm;
    for(ll i=1;i*i<=x&&i<=y;i++)///这里的循环结构要写的简洁，时间压力才小
    {
        if(x%i) continue;
        ans+=getm(i)*f(i,y/i);
        ll tp=x/i;
        if(tp<=y)   ans+=getm(tp)*f(tp,y/tp);
    }
    return ans*tm;
}
ll n,m;
int main()
{
    sieve();///筛前缀和和素数
    scanf("%lld%lld",&m,&n);
    printf("%lld\n",f(n,m));
    return 0;
}
/*
2000000000 2
*/