[로곡] 간단한 수학 문제 - 모비우스 반연 & 두교 체
21519 단어 적성 함수
문제풀이
이 방법은 비교적 복잡하지만 기교성이 있다.
이 방법은 비교적 직접적이고 간결하다.
게으르다
코드
#include
using namespace std;
const int N=8e6;
typedef long long ll;
ll n;int mod,iv2,iv6;
int p[N],tot,phi[N],ans,lim;
bool pri[N];
map<ll,int>mp;
inline int fp(int x,int y)
{
int re=1;
for(;y;y>>=1,x=(ll)x*x%mod) if(y&1) re=(ll)re*x%mod;
return re;
}
inline int ad(int x,int y){return ((ll)x+y)%mod;}
inline int dc(int x,int y){return ((ll)x-y+mod)%mod;}
inline void pre()
{
lim=(n<N)?(int)n:(N-1);int re,i,j;phi[1]=1;
for(int i=2;i<=lim;++i){
if(!pri[i]) p[++tot]=i,phi[i]=i-1;
for(j=1;j<=tot && (re=i*p[j])<=lim;++j){
pri[re]=true;
if(i%p[j]==0) {phi[re]=(ll)phi[i]*p[j]%mod;break;}
phi[re]=(ll)phi[i]*phi[p[j]]%mod;
}
}
for(i=2;i<=lim;++i) phi[i]=ad(phi[i-1],(ll)i*i%mod*(ll)phi[i]%mod);
}
inline int S(int x){return (ll)x*(x+1)%mod*(ll)iv2%mod;}
inline int Sp(int x){return (ll)x*(x+1)%mod*(ll)(((ll)x+x+1)%mod)%mod*(ll)iv6%mod;}
int F(ll x)
{
if(x<=lim) return phi[x];
if(mp[x]) return mp[x];
int re=S(x%mod);re=(ll)re*re%mod;
for(ll j,i=2;i<=x;i=j+1){
j=x/(x/i);
re=dc(re,(ll)dc(Sp(j%mod),Sp((i-1)%mod))*F(x/i)%mod);
}
return (mp[x]=re);
}
int main(){
ll i,j;int ss,vl,pr=0;
scanf("%d%lld",&mod,&n);pre();
iv2=fp(2,mod-2);iv6=fp(6,mod-2);
for(i=1;i<=n;i=j+1,pr=vl){
j=n/(n/i);ss=S((n/i)%mod);ss=(ll)ss*ss%mod;
vl=F(j);ans=ad(ans,(ll)ss*dc(vl,pr)%mod);
}
printf("%d",ans);
return 0;
}