HDU 6417 Rikka with APSP（min_25筛）

Description

给出一个$n$个点的完全图，其中$i,j$两点之间边的边权为最小的正整数$k$满足$ijk$为完全平方数，$d_{i,j}$为$i$点到$j$点的最短距离，求$\sum _{1\le a<b\le n}{d}_{a,b}$

Input

第一行一整数$T$表示用例组数，每组用例输入一整数$n(1\le T\le 50,1\le n\le 10^{10})$

Output

输出答案，结果模$998244353$

Sample Input

3
4
10
100

Sample Output

16
243
190371

Solution

以$S_{a,b}$表示$a,b$指数奇偶性不同的素数集合，若$|S_{a,b}|=0$则$d_{a,b}=1$，否则$d_{a,b}=\prod _{p\in S_{a,b}}p$，这两块分别求解

对于第一部分$\sum _{|S_{a,b}|=0}{d}_{a,b}$，此时$a,b$每个素因子对应的幂指数奇偶性均相同，我们拿出其中指数为奇数的素数乘起来，假设为$c$，则${\mu }^2(c)=1$($c$是无平方因子数等价于${\mu }^2(c)=1$)，同时$\frac{a}{c},\frac{b}{c}$均为完全平方数，开方后即为两个介于$[1,\lfloor \sqrt{\lfloor \frac{n}{c}\rfloor }\rfloor ]$的不同的数字，故有
$$\sum _{|S_{a,b}|=0}{d}_{a,b}=\sum _{c=1}^n{\mu }^2(c)C_{\lfloor \frac{n}{c}\rfloor }^2$$
由于$\lfloor \frac{n}{c}\rfloor$取值只有$O(\sqrt{n})$种，对于每种取值$x_i$，假设对应的$c$的区间为$[l_i,r_i]$，那么该部分对答案的贡献即为$C_{x_i}^2\sum _{j=l_i}^{r_i}{\mu }^2(j)$，注意到
$$\sum _{i=1}^n{\mu }^2(i)=\sum _{i=1}^n\sum _{d^2|i}\mu (d)=\sum _{d=1}^{\sqrt{n}}\mu (d)\lfloor \frac{n}{d^2}\rfloor$$
故在$O(\sqrt{n})$复杂度内即可求出${\mu }^2(i)$的前缀和，该部分总时间复杂度$O(\frac{n^{\frac{3}{4}}}{lo{g}_{2}n})$

对于第二部分$\sum _{|S_{a,b}|>0}{d}_{a,b}$，设$A_{n,p}$为$[1,n]$中$p$的指数为奇数的数字个数，其中$p$为素数，那么有
$$\sum _{|S_{a,b}|>0}{d}_{a,b}=\sum _{p}p\cdot A_{n,p}\cdot (n-A_{n,p})$$
对于$p\le \sqrt{n}$，有$A_{n,p}=\lfloor \frac{n}{p}\rfloor -A_{\lfloor \frac{n}{p}\rfloor ,p}$，在$O(lo{g}_{p}n)$复杂度内即可求出$A_{n,p}$

对于$p>\sqrt{n}$，有$A_{n,p}=\lfloor \frac{n}{p}\rfloor$，同理$A_p$的取值有$O(\sqrt{n})$种，对于每种取值对应$p$的区间$[l,r]$，我们需要求出该区间的素数和，$min\_25$筛即可，总时间复杂度也为$O(\frac{n^{\frac{3}{4}}}{lo{g}_{2}n})$

Code

#include<cstdio>
#include<algorithm>
using namespace std;
typedef long long ll;
#define mod 998244353
int add(int x,int y)
{
    x+=y;
    if(x>=mod)x-=mod;
    return x;
}
int mul(int x,int y)
{
    ll z=1ll*x*y;
    return z-z/mod*mod;
}
#define maxn 200005
#define maxm 1000006
int p[maxm],f[maxm],np=0,m=1000000,mu[maxm],s1[maxm],s[maxm];
void get_prime(int n)
{
    mu[1]=1,s[1]=1;
    for(int i=2;i<=n;i++)
    {
        s1[i]=s1[i-1];
        if(!f[i])p[++np]=i,mu[i]=-1,s1[i]=add(s1[i],i);
        s[i]=s[i-1]+(mu[i]!=0?1:0);
        for(int j=1;j<=np&&i*p[j]<=n;j++)
        {
            f[i*p[j]]=1,mu[i*p[j]]=-mu[i];
            if(i%p[j]==0)
            {
                mu[i*p[j]]=0;
                break;
            }
        }
    }
    np--;
}
ll val[maxn],n;
int nn,cnt;
void init()
{
    nn=1;
    while((ll)nn*nn<n)nn++;
    cnt=0;
    for(ll i=1;i<=n;i=n/(n/i)+1)val[++cnt]=n/i;
}
int ID(ll x)
{
    if(x>=nn)return n/x;
    return cnt-x+1;
}
int g1[maxn];
void Get_g(ll n)
{
    for(int i=1;i<=cnt;i++)
    {
        g1[i]=mul((val[i]-1)%mod,(val[i]+2)%mod);
        if(g1[i]&1)g1[i]+=mod;
        g1[i]/=2;
    }
    for(int j=1;j<=np;j++)
        for(int i=1;i<=cnt&&(ll)p[j]*p[j]<=val[i];i++)
        {
            int k=ID(val[i]/p[j]);
            g1[i]=add(g1[i],mod-mul(p[j],add(g1[k],mod-s1[p[j-1]])));
        }
    return ;
}
ll A(ll n,int p)
{
    if((ll)p*p>n)return n/p;
    return n/p-A(n/p,p);
} 
int S(ll n)
{
    if(n<=m)return s[n];
    int ans=0;
    for(int d=1;(ll)d*d<=n;d++)
        if(mu[d]==1)ans=add(ans,n/d/d%mod);
        else if(mu[d]==-1)ans=add(ans,mod-n/d/d%mod);
    return ans;
}
int main()
{
    get_prime(m);
    int T;
    scanf("%d",&T); 
    while(T--) 
    {
        scanf("%lld",&n);
        init();
        Get_g(n);
        int ans=0;
        for(int x=1;;x++)
        {
            if(val[x+1]>=nn)
                ans=add(ans,mul(add(g1[x],mod-g1[x+1]),mul(n/val[x]%mod,(n-n/val[x])%mod)));
            else
            {
                for(int i=1;i<=np&&p[i]<=val[x];i++)
                {
                    ll temp=A(n,p[i]);
                    ans=add(ans,mul(p[i],mul(temp%mod,(n-temp)%mod)));
                }
                break;
            }
        }
        int last=0,now;
        for(ll i=1,pre;i<=n;i=pre+1)
        {
            pre=n/(n/i);
            int temp=1;
            while((ll)temp*temp<=n/i)temp++;
            temp--;
            temp=(ll)temp*(temp-1)/2%mod;
            now=S(pre);
            ans=add(ans,mul(temp,add(now,mod-last)));
            last=now;
        }
        printf("%d\n",ans);
    } 
    return 0;
}