[BZOJ3944] Sum

All_fade_away

于 2019-02-22 22:01:24 发布

阅读量122

点赞数

分类专栏：数学-杜教筛文章标签：数学-杜教筛

本文链接：https://blog.csdn.net/xmr_pursue_dreams/article/details/87886209

版权

数学-杜教筛专栏收录该内容

2 篇文章 0 订阅

订阅专栏

[BZOJ3944] Sum

题目描述：

solution

裸的杜教筛。

唯一的坑点在于卡常。

似乎ans1和ans2都杜教筛超时了。

然而用杜教筛求出ans2，并用 $\mu$ 求出ans1不超时？！？。




#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int MAXN=3e6+50;
inline int read() 
{
	int f=1,x=0; char c=getchar();
	while (c<'0'||c>'9') { if (c=='-') f=-1; c=getchar(); }
	while (c>='0'&&c<='9') { x=(x<<3)+(x<<1)+(c^48); c=getchar(); }
	return x*f; 
}
map<int,int> Mu;
int prime[MAXN],vis[MAXN],n,pnum=0,inv2,inv6,mu[MAXN];
inline void init() 
{
    mu[1]=1;
    for(int i=2;i<MAXN;i++) 
    {
        if(!vis[i]) prime[++pnum]=i,mu[i]=-1;
        for(int t,j=1;j<=pnum&&i*prime[j]<MAXN;j++) 
        {
            vis[t=i*prime[j]]=1;
            if (!(i%prime[j])) {mu[t]=0;break;}
            mu[t]=-mu[i];
        }
    }
    for(int i=1;i<MAXN;i++) mu[i]=mu[i-1]+mu[i];
}
inline ll get_mu(ll n) 
{
    if (n<MAXN) return mu[n];
    if (Mu[n]) return Mu[n];
    ll T=2,res=1;
    while(T<=n)
    {
        ll pre=T;T=n/(n/T);
        res=res-1ll*(T-pre+1)*get_mu(n/T);T++;
    }
    return Mu[n]=res;
}
 
inline ll get_phi(ll n) 
{
    ll ans=0;
    for (ll T=1,pre;T<=n;T=pre+1)
    {
        pre=n/(n/T);
        ans+=1ll*(n/T)*(n/T+1)/2*(1ll*get_mu(pre)-1ll*get_mu(T-1));
    }
    return ans;
}
 
int main() 
{
    int t=read();
    init();
    while(t--) 
    {
        int n=read();
        printf("%lld %lld\n",get_phi(n),get_mu(n));
    }
    return 0;
}