結果

問題 No.1760 Setwise Coprime
ユーザー pockyny
提出日時 2021-11-20 18:01:18
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 93 ms / 2,000 ms
コード長 2,177 bytes
コンパイル時間 574 ms
コンパイル使用メモリ 69,204 KB
最終ジャッジ日時 2025-01-25 22:11:22
ジャッジサーバーID
（参考情報） 		judge2 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 36
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>

using namespace std;
typedef long long ll;
ll mod = 998244353;
ll pw(ll a,ll x){
    ll ret = 1;
    while(x){
        if(x&1) (ret *= a) %= mod;
        (a *= a) %= mod; x /= 2;
    }
    return ret;
}

ll gcd(ll a,ll b){
    if(a<b) swap(a,b);
    if(b==0) return a;
    return gcd(a%b,b);
}

ll lcm(ll a,ll b){return a*b/gcd(a,b);}
ll moe[200010],p[200010],f[200010],g[200010];
int main(){
    int i,j,n; cin >> n;
    for(i=1;i<=n;i++) moe[i] = 1;
    for(i=2;i<=n;i++){
        if(p[i]!=0) continue;
        moe[i] = -1;
        for(j=2;i*j<=n;j++){
            p[i*j] = i;
            if(j%i==0) moe[i*j] = 0;
            else moe[i*j] *= -1;
        }
    }
    for(i=2;i<=n;i++){
        if(moe[i]<0) moe[i] = mod - 1;
    }
    for(i=1;i<=n;i++) f[i] = pw(2,n/i)*moe[i]%mod;
    ll ans = 0,sum1 = 0,sum2 = 0;
    for(i=1;i<=n;i++) (sum1 += moe[i]) %= mod;
    for(i=1;i<=n;i++) (sum2 += f[i]) %= mod;
    (ans += sum1*sum1) %= mod;
    (ans += mod - (2*sum1*sum2)%mod) %= mod;
    /*for(i=1;i<=n;i++){
        for(j=1;j<=n;j++){
            ll d = lcm(i,j);
            int c1 = n/i - n/d,c2 = n/j - n/d,c3 = n/d;
            ll num = pw(2,c1 + c2)*pw(3,c3)%mod;
            (num *= moe[i]*moe[j]%mod) %= mod;
            (ans += num) %= mod;
        }
    }
    cout << ans << endl;
    return 0;*/ //kokomade atteru
    /*for(i=1;i<=n;i++){
        for(j=1;j<=n;j++){
            (g[lcm(i,j)] += f[i]*f[j]%mod) %= mod;
        }
    }*/
    /*for(i=1;i<=n;i++) cout << f[i] << " ";
    cout << endl;*/
    for(i=n;i>=1;i--){
        for(j=2*i;j<=n;j+=i){
            (f[j] += f[i]) %= mod;
        }
    }
    for(i=1;i<=n;i++) (f[i] *= f[i]) %= mod;
    for(i=1;i<=n;i++){
        for(j=2*i;j<=n;j+=i){
            (f[j] += mod - f[i]) %= mod;
        }
    }
    /*for(i=1;i<=n;i++) cout << f[i] << " ";
    cout << endl;
    for(i=1;i<=n;i++) cout << g[i] << " ";
    cout << endl;
    return 0;*/
    (sum2 *= sum2) %= mod;
    ll p = pw(4,mod - 2)*3%mod;//p = 3/4
    for(i=1;i<=n;i++) (sum2 += mod - f[i]) %= mod;
    (ans += sum2) %= mod;
    for(i=1;i<=n;i++){
        (ans += f[i]*pw(p,n/i)%mod) %= mod;
    }
    cout << ans << endl;
}
0