1. 程式人生 > >ACM-ICPC 2018 瀋陽網路賽 C Convex Hull (莫比烏斯+容斥)

ACM-ICPC 2018 瀋陽網路賽 C Convex Hull (莫比烏斯+容斥)

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);
    }
}