【XSY2719】prime 莫比乌斯反演-CSDN博客

本文链接：https://blog.csdn.net/ez_yww/article/details/79059172

题目描述

　　设 $f(i)$ 为 $i$ 的不同的质因子个数，求 $\sum_{i=1}^n2^{f(i)}$

　　 $n\leq{10}^{12}$

题解

　　考虑 $2^{f(i)}$ 的意义：有 $f(i)$ 总因子，每种可以分给两个人中的一个。那么就有 $2^{f(i)}=\sum_{d|i}[\gcd(d,\frac{i}{d})=1]$

　　然后就是简单莫比乌斯反演了。

s = \sum i = 1 n \sum d | i [gcd (d, i d) = 1] = \sum i = 1 n \sum d | i \sum j | d && j | i d μ (j) = \sum i = 1 n \sum j 2 | i g (i j 2) μ (j) = \sum j = 1 n \sqrt μ (j) \sum i = 1 ⌊ n j 2 ⌋ g (i) = \sum j = 1 n \sqrt μ (j) \sum i = 1 ⌊ n j 2 ⌋ ⌊ n j 2 i ⌋

$\begin{align} s&=\sum_{i=1}^n\sum_{d|i}[\gcd(d,\frac{i}{d})=1]\\ &=\sum_{i=1}^n\sum_{d|i}\sum_{j|d\text{&&}j|\frac{i}{d}}\mu(j)\\ &=\sum_{i=1}^n\sum_{j^2|i}g(\frac{i}{j^2})\mu(j)\\ &=\sum_{j=1}^\sqrt{n}\mu(j)\sum_{i=1}^{\lfloor\frac{n}{j^2}\rfloor}g(i)\\ &=\sum_{j=1}^\sqrt{n}\mu(j)\sum_{i=1}^{\lfloor\frac{n}{j^2}\rfloor}\lfloor\frac{n}{j^2i}\rfloor \end{align}$
　　时间复杂度：

O(n√logn) $O(\sqrt n\log n)$

代码

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long ll;
const ll p=998244353;
ll gao(ll x)
{
    ll s=0;
    ll i,j;
    for(i=1;i<=x;i=j+1)
    {
        j=x/(x/i);
        s+=(x/i)*(j-i+1);
    }
    return s;
}
int b[1000010];
int pri[1000010];
int cnt;
int miu[1000010];
int main()
{
    ll i,j;
    miu[1]=1;
    for(i=2;i<=1000000;i++)
    {
        if(!b[i])
        {
            pri[++cnt]=i;
            miu[i]=-1;
        }
        for(j=1;j<=cnt&&i*pri[j]<=1000000;j++)
        {
            b[i*pri[j]]=1;
            if(i%pri[j]==0)
            {
                miu[i*pri[j]]=0;
                break;
            }
            miu[i*pri[j]]=-miu[i];
        }
    }
    ll ans=0;
    ll n;
    scanf("%lld",&n);
    for(i=1;i*i<=n;i++)
        ans=(ans+miu[i]*gao(n/(i*i)))%p;
    ans=(ans+p)%p;
    printf("%lld\n",ans);
    return 0;
}