bzoj4916 神犇和蒟蒻（杜教筛）

最新推荐文章于 2020-09-11 18:43:43 发布

live4m

最新推荐文章于 2020-09-11 18:43:43 发布

阅读量122

点赞数

本文链接：https://blog.csdn.net/weixin_44178736/article/details/108304901

版权

题意：

给定n，要求计算：
$\sum_{i=1}^nμ(i^2)$
$\sum_{i=1}^nφ(i^2)$
答案对1e9+7取模

数据范围：n<=1e9

解法：

第一问的答案一定是1，因为只有1²没有平方质因子，其他都有，
根据莫比乌斯函数定义，有平方质因子的数的莫比乌斯函数值为0

那么剩下的就是第二问了：
欧拉函数： $φ(n)=\frac {n}{p_{1}p_{2}...p_{m}}(p_{1}-1)(p_{2}-1)..(p_{m}-1)$
因为n²的质因子在n中已经出现过了，那么： $φ(n^2)=\frac {n^2}{p_{1}p_{2}...p_{m}}(p_{1}-1)(p_{2}-1)..(p_{m}-1)$
可以把分子的一个n提出来，那么： $φ(n^2)$ = $n * φ (n)$
问题变为计算 $\sum_{i=1}^ni*φ(i)$

推导：
$\sum_{i=1}^ni*φ(i)$
$令 f = i * φ (i), 选择 g = i d (i) = i$
$那么(f*g)(n)=\sum_{d|n}f(d)*g(\frac nd)$
$=\sum_{d|n}d*φ(d)*\frac nd$
$=n*\sum_{d|n}φ(d)$
$n^2$
最后一步用了欧拉函数的性质 $\sum_{d|n}φ(d)=n$

综上： $f*g)(i)=i^2$

杜教筛递推式： $g(1)S(n)=\sum_{i=1}^n(f*g)(i)-\sum_{i=2}^ng(i)S(\lfloor \frac ni \rfloor)$
代入得： $S(n)=\sum_{i=1}^ni^2-\sum_{i=2}^ni*S(\lfloor \frac ni \rfloor)=\frac {n*(n+1)*(2n+1)}{6}-\sum_{i=2}^ni*S(\lfloor \frac ni \rfloor)$

code：

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int maxm=1e6+5;
const int mod=1e9+7;
const int inv6=166666668;
ll phi[maxm];
ll notprime[maxm];
ll prime[maxm],cnt;
ll sum_p[maxm];
void init(){
    phi[1]=1;
    for(int i=2;i<maxm;i++){
        if(!notprime[i]){
            prime[cnt++]=i;
            phi[i]=i-1;
        }
        for(int j=0;j<cnt;j++){
            if(prime[j]*i>=maxm)break;
            notprime[prime[j]*i]=1;
            phi[prime[j]*i]=phi[i]*(i%prime[j]?prime[j]-1:prime[j]);
            if(i%prime[j]==0)break;
        }
    }
    for(int i=1;i<maxm;i++){
        sum_p[i]=(sum_p[i-1]+phi[i]*i%mod)%mod;
    }
}
map<ll,ll>mp_p;
ll cal(ll x){
    return x*(x+1)/2;
}
ll S_p(ll x){
    if(x<maxm)return sum_p[x];
    if(mp_p[x])return mp_p[x];
    ll ans=x*(x+1)%mod*(2*x+1)%mod*inv6%mod;
    for(ll i=2,j;i<=x;i=j+1){
        j=x/(x/i);
        ll temp=(cal(j)-cal(i-1))%mod;
        ans-=temp*S_p(x/i)%mod;
        ans%=mod;
    }
    ans=(ans%mod+mod)%mod;
    return mp_p[x]=ans;
}
signed main(){
    init();
    ll n;scanf("%lld",&n);
    printf("1\n%lld",S_p(n));
    return 0;
}