Libre OJ #6053.简单的函数

简单的函数

传送门

Description

Data Constraints

$n\leq10^{10}$

Solution

Min_25筛版子题。
Min_25筛（点一下玩一年）

先预处理出$\sqrt n$内的所有质数并排好序，第$i$个质数记为$p_i$,设共有$v$个质数。

设
$g(n,j)=\sum_{i=2}^ni[i是质数或i的最小质因子大于p_j]$
$h(n,j)=\sum_{i=2}^n[i是质数或i的最小质因子大于p_j]$
那我们有

$$g(n,j)=\left\{ \begin{array}{} g(n,j-1)-p_j[g(\lfloor\frac{n}{p_j}\rfloor,j-1)-g(p_{j-1},j-1)]& &{p_j^2\leq n}\\ g(n,j-1)& &{p_j^2\geq n} \end{array}\right.$$
$$h(n,j)=\left\{ \begin{array}{} h(n,j-1)-[h(\lfloor\frac{n}{p_j}\rfloor,j-1)-h(p_{j-1},j-1)]& &{p_j^2\leq n}\\ h(n,j-1)& &{p_j^2\geq n} \end{array}\right.$$

那显然有$g(n,v)-h(n,v)=\sum_{i=2}^nF(i)[i是质数]$，记$g(n,v)-h(n,v)=f(n)$

接下来就能按照$Min\_25$筛的套路求出该函数前$n$项的总和了。

设

$$S(n,i)=\sum_{i=2}^nF(i)[i的最小质因子不小于p_i]$$

$$S(n,i)=f(n)-f(p_{i-1})+\sum_{j=i}^v\ \ \ \sum_{e\geq 1且p_j^{e+1}\leq n}(p_j⊕e)S(\lfloor\frac{n}{p_j^e}\rfloor,j+1)+(p_j⊕(e+1))$$

那所求值显然为$S(n,1)+F(1)$，递归求即可，这样就做完了，时间复杂度$O(\frac{n^{\frac{3}{4}}}{log\sqrt n})$。

Code

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>

#define fo(i,j,l) for(int i=j;i<=l;++i)
#define fd(i,j,l) for(int i=j;i>=l;--i)

using namespace std;
typedef long long ll;
const ll N=4e5,mo=1e9+7;

ll n,P,oo,m;
int s[N];
bool bz[N];
int id1[N],id2[N];
ll g[N],w[N],zh[N],h[N];

inline ll ask(ll x,int k)
{
    if(x<=1||s[k]>x)return 0;
    int wz=x<=P?id1[x]:id2[n/x];
    ll ans=(g[wz]-h[wz]-zh[k-1]+k-1+mo+mo)%mo;
    if(k==1)ans+=2;
    for(int i=k;i<=oo&&(ll)s[i]*s[i]<=x;++i)
    for(ll l=s[i],e=1;l*s[i]<=x;l=l*s[i],++e)
    ans=(ans+ask(x/l,i+1)*(s[i]^e)+(s[i]^(e+1)))%mo;
    return ans;
}

int main()
{
    cin>>n;
    P=sqrt(n);
    fo(i,2,P){
        if(!bz[i])s[++oo]=i,zh[oo]=(zh[oo-1]+i)%mo;
        fo(j,1,oo)if(i*s[j]<=P){
            bz[i*s[j]]=true;
            if(i%s[j]==0)break;
        }else break;
    }
    ll l=1,r;
    while(l<=n){
        r=n/(n/l);
        if(n/l<=P)id1[n/l]=++m;else id2[r]=++m;
        ll len=n/l;
        w[m]=len; ll y=len%mo;
        g[m]=(y*(y+1)/2-1)%mo; h[m]=(len-1)%mo;
        l=r+1;
    }
    fo(i,1,oo)
    fo(j,1,m)if((ll)s[i]*s[i]>w[j])break;
    else{
        int op=w[j]/s[i]<=P?id1[w[j]/s[i]]:id2[n/(w[j]/s[i])];
        g[j]=(g[j]-s[i]*(g[op]-zh[i-1])%mo+mo)%mo;
        h[j]=(h[j]-(h[op]-(i-1))+mo)%mo;
    }
    ll ans=(ask(n,1)+1)%mo;
    cout<<ans<<endl;
}