[数位DP 莫比乌斯反演] 2017 计蒜之道 复赛 A. 阿里云秘钥池

大概可以推出来转移是
$f_x=\sum_{d|x}\mu(d)\sum_{i=1}^{\lfloor {P-1\over d}\rfloor} g_{id}=\sum_{d|x}\mu(d)G(d)$
转移是$O(n\ln n)$的
然后就直接数位dp咯

#include<cstdio>
#include<cstdlib>
#include<algorithm>
using namespace std;
typedef long long ll;

#define read(x) scanf("%d",&(x))

const ll PP=2e18;

inline void add(ll &x,ll y){
  x+=y; if (x>=PP) x-=PP;
}

const int maxn=100005;
int prime[maxn],vst[maxn];
int num,mu[maxn];

inline void Pre(int n){
  mu[1]=1;
  for (int i=2;i<=n;i++){
    if (!vst[i]) prime[++num]=i,mu[i]=-1;
    for (int j=1;j<=num && (ll)i*prime[j]<=n;j++){
      vst[i*prime[j]]=1;
      if (i%prime[j]==0){
    mu[i*prime[j]]=0; break;
      }else{
    mu[i*prime[j]]=-mu[i];
      }
    }
  }
}

int P;

int a[70],m;
ll f[70][maxn],g[70][maxn];

inline void calc(ll *f,ll *g){
  for (int i=0;i<P;i++) g[i]=0;
  for (int i=1;i<P;i++)
    for (int j=i;j<P;j+=i)
      add(g[i],f[j]);
}

inline ll Solve(ll n){
  m=0;
  while (n) a[++m]=n%P,n/=P;

  int flag=1;
  for (int i=0;i<P;i++) f[m][i]=0;
  for (int i=1;i<a[m];i++) add(f[m][i],1);
  calc(f[m],g[m]);

  for (int t=m-1;t;t--){
    for (int i=0;i<P;i++) f[t][i]=0;
    for (int i=1;i<P;i++)
      if (mu[i])
    for (int j=i;j<P;j+=i)
      add(f[t][j],mu[i]==1?g[t+1][i]:(PP-g[t+1][i])%PP);
    for (int i=1;i<P;i++)
      add(f[t][i],1);
    if (flag){
      for (int i=1;i<a[t];i++)
    if (__gcd(i,a[t+1])==1)
      add(f[t][i],1);
      if (a[t]==0 || __gcd(a[t],a[t+1])!=1)
    flag=0;
    }
    calc(f[t],g[t]);
  }

  ll ret=0;
  for (int i=1;i<P;i++)
    add(ret,f[1][i]);
  return ret;
}

int main(){
  int T; ll L,R;
  freopen("t.in","r",stdin);
  freopen("t.out","w",stdout);
  read(T); Pre(100000);
  while (T--){
    scanf("%lld%lld%d",&L,&R,&P);
    printf("%lld\n",Solve(R+1)-Solve(L));
  }
  return 0;
}