ACM-ICPC 2018 瀋陽網路賽 C Convex Hull (莫比烏斯+容斥)
阿新 • • 發佈:2018-12-12
long long 爆的我心痛,現在編譯器都支援128位的了。。。
神奇,電腦編譯器都過不了(老了),交上去ac了,就是複雜度還有點高,應該不是正解,不過又get到了新知識,
不過也有的巨佬說是餘數太大的緣故,所以中間採用 int128
很菜,容斥推導那塊想了半天才恍惚明白過來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); } }