BZOJ3739 : DZY loves math VIII

显然当且仅当$gcd(i,j)=1$时才对答案有贡献，化简得

[egin{eqnarray*}
ans&=&sum_{i=1}^nsum_{j=1}^imu(ij)[gcd(i,j)=1]\
&=&sum_{i=1}^nsum_{j=1}^imu(ij)sum_{d|i,d|j}mu(d)\
&=&sum_{i=1}^nmu(i)sum_{d|i}mu(d)sum_{j=1}^{frac{i}{d}}mu(dj)
end{eqnarray*}]

设

[S(i,d)=sum_{j=1}^{frac{i}{d}}mu(dj)]

则

[ans=sum_{i=1}^nmu(i)sum_{d|i}mu(d)S(i,d)]

当且仅当$i$是square-free的时候，才对答案有贡献。此时将$i$分解质因数，然后暴力搜索所有约数，一边搜索一边更新$S(i,d)$以及$ans$即可。

#include<cstdio>
#define N 10000010
int T,n,i,j,A,B,q[1010],tot,p[700000],v[N],s[N];char mu[N];short f[N];
void dfs(int x,int y){
  if(x==tot){
    f[y]+=A;
    if(mu[y]>0)B+=f[y];else B-=f[y];
    return;
  }
  dfs(x+1,y*p[x]),dfs(x+1,y);
}
int main(){
  scanf("%d",&T);
  for(i=1;i<=T;i++){
    scanf("%d",&q[i]);
    if(q[i]>n)n=q[i];
  }
  for(mu[1]=v[1]=1,i=2;i<=n;i++){
    if(!v[i])p[tot++]=v[i]=i,mu[i]=-1;
    for(j=0;j<tot;j++){
      if(i*p[j]>n)break;
      v[i*p[j]]=p[j];
      if(i%p[j])mu[i*p[j]]=-mu[i];else break;
    }
  }
  for(i=1;i<=n;i++){
    s[i]=s[i-1];
    if(mu[i]){
      for(tot=0,j=i;j>1;j/=v[j])p[tot++]=v[j];
      A=mu[i],B=0,dfs(0,1);
      if(mu[i]>0)s[i]+=B;
      if(mu[i]<0)s[i]-=B;
    }
  }
  for(i=1;i<=T;i++)printf("%d
",s[q[i]]);
  return 0;
}