51Nod 1239 欧拉函数前n项和 杜教筛

http://www.51nod.com/Challenge/Problem.html#!#problemId=1239

AC代码

#include <bits/stdc++.h>
#define pb push_back
#define mp make_pair
#define fi first
#define se second
#define all(a) (a).begin(), (a).end()
#define fillchar(a, x) memset(a, x, sizeof(a))
#define huan printf("\n");
#define debug(a,b) cout<<a<<" "<<b<<" ";
using namespace std;
const int maxn=1e6+,inf=0x3f3f3f3f;
typedef long long ll;
const ll mod = ;
typedef pair<int,int> pii;
int check[maxn],prime[maxn],phi[maxn],sum[maxn];
void Phi(int N)//莫比乌斯函数线性筛
{
    int pos=;sum[]=phi[]=;
    for(int i =  ; i <= N ; i++)
    {
        if (!check[i])
            prime[pos++] = i,phi[i]=i-;
        for (int j =  ; j < pos && i*prime[j] <= N ; j++)
        {
            check[i*prime[j]] = ;
            if (i % prime[j] == )
            {
                phi[i*prime[j]]=phi[i]*prime[j];
                break;
            }
            else
                phi[i*prime[j]]=phi[i]*(prime[j]-);
        }
        sum[i]=(sum[i-]+phi[i])%mod;
    }
}
unordered_map<ll,ll> ma;
ll inv2=;
ll solve(ll n)
{
    if(n<=1e6)
        return sum[n];
    else if(ma.count(n))
        return ma[n];
    ll temp = ((n%mod)*((n+)%mod)%mod)*inv2%mod;
    for(ll i=,j;i<=n;i=j+)
    {
        j=n/(n/i);
        temp = (temp-solve(n/i)*(j-i+)%mod+mod)%mod;
    }
    return ma[n]=temp;
}
int main()
{
    ll n;
    Phi(1e6);
    scanf("%lld",&n);
    printf("%lld\n",solve(n));
}