HDU6588 经典gcd求和

需要求

$$\sum _{i=1}^n\gcd (\lfloor \sqrt[3]{i}\rfloor ,i)\mathrm{m}\mathrm{o}\mathrm{d}998244353n=10^{21}$$

思路

令$m=\lfloor \sqrt[3]{n}\rfloor -1$，

则${\sum }_{i=1}^{n}gcd(floor(\sqrt[3]{i}),i)={\sum }_{i=1}^m\sum _{j=i^3}^{(i+1{)}^3-1}gcd(i,j)+\sum _{i=(m+1{)}^3}^{n}gcd(m+1,i)$

考虑一个函数（前缀和，差分后即为上面部分所求，这个想法也比较自然），$f(n)={\sum }_{i=1}^{n}gcd(m,i)$
$$\begin{gathered} f(n)=\sum _{i=1}^{n}gcd(m,i) \\ =\sum _{i=1}^n\sum _{d|gcd(m,i)}\varphi (d) \\ =\sum _{d|m}\varphi (d)\ast \lfloor \frac{n}{d}\rfloor \end{gathered}$$
则 :
注意下面的，$m=\lfloor \sqrt[3]{n}\rfloor -1$
$$\begin{gathered} \sum _{i=1}^{n}gcd(floor(\sqrt[3]{i}),i)=\sum _{i=1}^m\sum _{j=i^3}^{(i+1{)}^3-1}gcd(i,j)+\sum _{i=(m+1{)}^3}^{n}gcd(m+1,i) \\ =\sum _{i=1}^m\sum _{d|i}\varphi (d)\ast \left[\lfloor \frac{(i+1{)}^3-1}{d}\rfloor -\lfloor \frac{i^3-1}{d}\rfloor \right]+\sum _{d|m+1}\varphi (d)\ast \left[\lfloor \frac{n}{d}\rfloor -\lfloor \frac{(m+1{)}^3-1}{d}\rfloor \right] \\ =\sum _{d=1}^m\varphi (d)\ast \sum _{i=1}^{\lfloor m/d\rfloor }\left[\lfloor \frac{(id+1{)}^3-1}{d}\rfloor -\lfloor \frac{(id{)}^3-1}{d}\rfloor \right]+\sum _{d|m+1}\varphi (d)\ast \left[\lfloor \frac{n}{d}\rfloor -\lfloor \frac{(m+1{)}^3-1}{d}\rfloor \right] \\ =\sum _{d=1}^m\varphi (d)\ast \sum _{i=1}^{\lfloor m/d\rfloor }\left[3d{i}^2+3i+1(注意这里是下取整)\right]+\sum _{d|m+1}\varphi (d)\ast \left[\lfloor \frac{n}{d}\rfloor -\lfloor \frac{(m+1{)}^3-1}{d}\rfloor \right] \end{gathered}$$
因此可以计算 ，时间复杂度$O(1e7+T\ast (1e7+\sqrt{1e7})$ 其中$T$为数据组数，最左边的$1e7$为预处理欧拉函数的时间。

本题的关键在于推导的倒数第二行变换贡献，从枚举$i$变为枚举$d$，使得复杂度降了下来

Code

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef __int128 lll;
const int mod = 998244353;
const int inv2 = 499122177;
const int inv6 = 166374059;
const int maxn=1e7+10;
template <class T>
void read(T &x) {
	static char ch;static bool neg;
	for(ch=neg=0;ch<'0' || '9'<ch;neg|=ch=='-',ch=getchar());
	for(x=0;'0'<=ch && ch<='9';(x*=10)+=ch-'0',ch=getchar());
	x=neg?-x:x;
}
lll n;
int tot;
bool valid[maxn];//is prime or not
int phi[maxn];// phi
int ans[maxn];// prime
void getprime(int n)
{
    tot=0;
    phi[1]=1;
    memset(valid,true,sizeof(valid));
    for(int i=2;i<=n;++i){
        if(valid[i]){
            tot++;
            ans[tot]=i;
            phi[i]=i-1;//i为素数
        }
        //下面的主角是小于等于i的每个质数
        for(int j=1;(j<=tot) && (i*ans[j]<=n);++j){
            valid[i*ans[j]]=false;
            if(i%ans[j]==0){//ans[j]是i的素因子
                phi[i*ans[j]]=phi[i]*ans[j];
                break;//如果整除就break;
            }
            else phi[i*ans[j]]=phi[i]*(ans[j]-1);
        }
    }
}
lll pow3(lll x)
{
    return x*x*x;
}
//n(n+1)(2n+1)/6
ll solve(int d, ll m){
    ll up = m/d;
    ll ans1 = (1+up)*up%mod*inv2%mod;
    ll ans2 = d*up%mod*(up+1)%mod*(2*up+1)%mod*inv6%mod;
    return ((ans1+ans2)*3 + up)%mod;
}
int main()
{
    getprime(1e7);
    int t;
    cin>>t;
    while(t--)
    {
        read(n);
        int m = pow(n,1.0/3) - 1e5;
        for(;pow3(m)<=n;m++);
        m-=2;
        ll ans = 0;
        for(int d=1;d<=m;++d){
            ans=(ans + 1LL*phi[d]*solve(d,m)%mod)%mod;
        }
        ll res = 0;
        int now=m+1;
        for(int i=1;i*i<=now;++i){
            if(now%i==0){//i now/i
                res = (res + 1LL*phi[i]*((n/i-(pow3(m+1)-1)/i)%mod) )%mod;
                if(i*i!=now) res = (res + 1LL*phi[now/i]*((n/(now/i)-(pow3(m+1)-1)/(now/i))%mod) )%mod;
            }
        }
        ans = (ans + res)%mod;
        cout<<ans<<endl;
    }
    return 0;
}