long long 爆的我心痛,现在编译器都支持128位的了。。。
神奇,电脑编译器都过不了(老了),交上去ac了,就是复杂度还有点高,应该不是正解,不过又get到了新知识,
不过也有的巨佬说是余数太大的缘故,所以中间采用 int128
参考博文:https://blog.csdn.net/qkoqhh/article/details/82532516#commentsedit
很菜,容斥推导那块想了半天才恍惚明白过来QAQ
就不写推到了,上面博客写的很清楚了。
#include<bits/stdc++.h>
using namespace std;
typedef unsigned long long ll;
const ll maxn=1e5+10;
pair<__int128,__int128> p[maxn];
ll mod;
bool vis[maxn];
int prime[maxn];
int mu[maxn];
void init(){
mu[1]=1;
int tot=0;
for(int i=2;i<maxn;i++){
if(!vis[i]){
prime[tot++]=i;
mu[i]=-1;
}
for(int j=0;j<tot;j++){
if(i*prime[j]>maxn) break;
vis[i*prime[j]]=true;
if(i%prime[j]==0){
mu[i*prime[j]]=0;
break;
}else{
mu[i*prime[j]]=-mu[i];
}
}
}
}
ll pingfang(ll n){
__int128 x=n,y=n+1,z=2*n+1;
if(x%2==0) x/=2;
else if(y%2==0) y/=2;
else z/=2;
if(x%3==0) x/=3;
else if(y%3==0) y/=3;
else z/=3;
return (x%mod*y%mod*z%mod);
}
ll lifang(ll n){
__int128 x=n,y=n+1;
if(x%2==0) x/=2;
else y/=2;
return x%mod*x%mod*y%mod*y%mod;
}
int main(){
ll n;
init();
while(scanf("%lld%lld",&n,&mod)!=EOF){
for(ll i=0;i*i<=n;i++){
p[i].first=i%mod*i%mod*i%mod*i%mod;
p[i].second=i%mod*i%mod*i%mod*i%mod*i%mod*i%mod;
}
ll ans=0;
for(ll i=1;i*i<=n;i++){
ans+=mod+(n+1)%mod*p[i].first%mod*pingfang(n/(i*i))%mod*mu[i]-p[i].second%mod*lifang(n/(i*i))%mod*mu[i];
ans=(ans+mod)%mod;
}
printf("%lld\n",ans);
}
}