bzoj3944 Sum(杜教筛入门/欧拉函数+莫比乌斯函数)

题目

T(T<=10)组样例，每次给出一个非负整数n(0<=n<=2的31次方-1)

求和

思路来源

https://blog.csdn.net/skywalkert/article/details/50500009 唐老师博客例题

题解

n为2e9，预处理前的值，大概前2e6

以下图片来自唐老师博客：

卡常题目，得把一切能开int的地方都开int才能过

也就T+WA了10来发吧……

①n开int，n+1就爆ll了，

②map可以优化成unordered_map或手写hash

③读入挂read可以再缩时间

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<map>
using namespace std;
typedef long long ll;
const int maxn=3e6+10;
bool ok[maxn];
int prime[maxn],mu[maxn],cnt;
ll phi[maxn];
int T,n;
map<int,ll>sphi;
map<int,int>smu;
void sieve()
{
    phi[1]=mu[1]=1;
    for(ll i=2;i<maxn;++i)
    {
        if(!ok[i])
        {
            prime[cnt++]=i;
            phi[i]=i-1;
            mu[i]=-1;
        }
        for(int j=0;j<cnt;++j)
        {
            ll k=i*prime[j];
            if(k>=maxn)break;
            ok[k]=1;
            if(i%prime[j]==0)
            {
                phi[k]=phi[i]*prime[j];
                mu[k]=0;
                break; 
            }
            else
            {
             phi[k]=phi[i]*(prime[j]-1);
             mu[k]=-mu[i];
            }
        }
    }
    for(int i=2;i<maxn;++i)
    {
        phi[i]+=phi[i-1];
        mu[i]+=mu[i-1];
    }
}
ll djsphi(int n)
{
    if(n<maxn)return phi[n];
    if(sphi.count(n))return sphi[n];
    ll ans=1ll*n*(1ll*n+1)/2;
    for(int l=2,r;l<=n;l=r+1)
    {
        r=n/(n/l);
        ans=ans-(r-l+1)*djsphi(n/l);
        if(r==n)break;
    }
    return sphi[n]=ans;
}
int djsmu(int n)
{
    if(n<maxn)return mu[n];
    if(smu.count(n))return smu[n];
    int ans=1;
    for(int l=2,r;l<=n;l=r+1)
    {
        r=n/(n/l);
        ans=ans-(r-l+1)*djsmu(n/l);
        if(r==n)break;
    }
    return smu[n]=ans;
}
int main()
{
    sieve();
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d",&n);
        printf("%lld %d\n",djsphi(n),djsmu(n));
    }
    return 0;
}