杭电多校第一场1011Function（莫比乌斯反演）

杭电多校第一场1011Function（莫比乌斯反演）

题目大意

求
$$\sum _{i=1}^{n}gcd(\lfloor \sqrt[3]{i}\rfloor ,i)mod998244353$$

解题思路

以下极大一部分来自出题人的题解，仅仅对自己原来觉得有些跳得部分进行了补充，感谢出题人

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

则原式等于

$$\begin{array}{rl}& \sum _{i=1}^n\gcd (\lfloor \sqrt[3]{i}\rfloor ,i)\\ =& \sum _{i=\lfloor \sqrt[3]{n}{\rfloor }^3}^n\gcd (\lfloor \sqrt[3]{n}\rfloor ,i)+\sum _{a=1}^r\sum _{i=a}^{(a+1{)}^3-1}\gcd (a,i)\end{array}$$
其中
$$\begin{array}{rl}\sum _{i=1}^n\gcd (a,i)=& \sum _{d|a}d\sum _{i=1}^n[\gcd (a,i)=d]\\ =& \sum _{d|a}d\sum _{i=1}^{\lfloor \frac{n}{d}\rfloor }[gcd(i,\frac{a}{d})=1]\\ =& \sum _{d|a}d\sum _{i=1}^{\lfloor \frac{n}{d}\rfloor }\sum _{f|gcd(i,\frac{a}{d})}\mu (f)\\ =& \sum _{d|a}d\sum _{f|\frac{a}{d}}\mu (f)\lfloor \frac{n}{df}\rfloor \\ =& \sum _{df|a}\lfloor \frac{n}{df}\rfloor \sum _{d|df}d\mu (\frac{df}{d})\end{array}$$
而根据迪利克雷卷积中$\phi =\mu \ast e$

有${\sum }_{d|df}d\mu (\frac{df}{d})=\varphi (df)$，接下来$\sqrt{a}$枚举a的因数则可以得出第一部分的答案

第二个和式可以化为
$$\begin{array}{rl}& \sum _{a=1}^r\sum _{T|a}(\lfloor \frac{(a+1{)}^3-1}{T}\rfloor -\lfloor \frac{a^3-1}{T}\rfloor )\phi (T)\\ =& \sum _T\phi (T)\sum _{b=1}^{\lfloor \frac{r}{T}\rfloor }(\lfloor \frac{(bT+1{)}^3-1}{T}\rfloor -\lfloor \frac{(bT{)}^3-1}{T}\rfloor )\end{array}$$
而
$$\begin{array}{rl}& \lfloor \frac{(bT+1{)}^3-1}{T}\rfloor -\lfloor \frac{(bT{)}^3-1}{T}\rfloor \\ =& \lfloor b^3{T}^2+3b^{2}T+3b\rfloor -\lfloor b^3{T}^2-\frac{1}{T}\rfloor \\ =& 3T{b}^2+3b+1\end{array}$$
因此
$$\begin{array}{rl}& \sum _{a=1}^r\sum _{i=a}^{(a+1{)}^3-1}\gcd (a,i)\\ =& \sum _{T=1}^r\phi (T)[(3T\sum _{b=1}^{\lfloor \frac{r}{T}\rfloor }{b}^2)+(3\sum _{b=1}^{\lfloor \frac{r}{T}\rfloor }b)+\lfloor \frac{r}{T}\rfloor ]\end{array}$$
可以$O(\sqrt[3]{n})$求出。
总复杂度$O(T\sqrt[3]{n})$。

$n$比较小的情况可能会出问题，注意要适当特判。也有其它的算法，只要是线性应该都是能过的。把$\sqrt[3]{n}$出到$10^7$是为了卡非线性做法。

AC代码

#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
const double eps=1e-6;
const int mod=998244353;
const int size=1e7+5;
int p[size];bool prime[size];
int phi[size],tot;
int quick_pow(int a,int b){int ans=1;while(b){if(b&1)ans=1LL*ans*a%mod;a=1LL*a*a%mod;b>>=1;} return ans;}
int inv6,inv2;
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;
}
void init()
{
    inv6=quick_pow(6,mod-2);inv2=quick_pow(2,mod-2);
    phi[1]=1;
    for(int i=1;i<size;i++) prime[i]=true;
    for(int i=2;i<size;i++)
    {
        if(prime[i]) 
        p[++tot]=i,phi[i]=i-1;
        for(int j=1;j<=tot&&p[j]*i<size;j++)
        {
            prime[i*p[j]]=false;
            if(i%p[j]==0)
            {
                phi[i*p[j]]=phi[i]*p[j];
                break;
            }
            else phi[i*p[j]]=phi[i]*phi[p[j]];
        }
    }
}
LL sum2(int n)
{
    return 1LL*n*(n+1)%mod*(2*n+1)%mod*inv6%mod;
}
LL sum1(int n)
{
    return 1LL*n*(1+n)%mod*inv2%mod;
}
int main()
{
    init();
    int t;
    scanf("%d",&t);
    while(t--)
    {
        __int128 n;
        read(n);
        LL a=(pow(n,1.0/3)+eps);
        LL ans=0;
        for(int i=1;i<=int(sqrt(a)+eps);i++)
        {
            LL t=i%mod;
            LL tmp=(((__int128)(n/t)-((__int128)a*a*a-1)/t)%mod+mod)%mod;
            if(a%i) continue;
            ans=(ans+1LL*phi[t]%mod*tmp)%mod;
            if(i*i==a) continue;
            t=a/i%mod;
            tmp=(((__int128)(n/t)-((__int128)a*a*a-1)/t)%mod+mod)%mod;
            ans=(ans+1LL*phi[t]%mod*tmp)%mod;
        }
        LL r=a-1;
        for(LL T=1;T<=r;T++)
        {
            ans=(ans+1LL*phi[T]*((3*T*sum2(r/T)%mod+3*sum1(r/T)+r/T)%mod)%mod)%mod;
        }
        printf("%lld\n",ans);
    }
}