結果

問題 No.732 3PrimeCounting
コンテスト
ユーザー てんぷら
提出日時 2023-09-22 00:01:31
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 2,141 ms / 3,000 ms
コード長 2,368 bytes
コンパイル時間 5,124 ms
コンパイル使用メモリ 315,220 KB
実行使用メモリ 24,028 KB
最終ジャッジ日時 2024-07-07 16:57:42
合計ジャッジ時間 37,011 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 89
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <atcoder/all>
#include <bits/stdc++.h>
using ll = long long;
#define rep(i, n) for(ll i = 0; i < (ll)(n); i++)
#define REP(i, m, n) for(ll i = (ll)(m); i < (ll)(n); i++)
using namespace std;
using namespace atcoder;
using mint = modint998244353;

tuple<vector<ll>, vector<ll>, vector<ll>> solve(int l, int r, vector<int> &f) {
    if(r - l == 1) {
        return {{0}, {0}, {0}};
    }
    int m = (l + r) / 2;
    auto [l12, l23, l123] = solve(l, m, f);
    auto [r12, r23, r123] = solve(m, r, f);
    vector<ll> a12(2 * (r - l) - 1), a23(2 * (r - l) - 1), a123(3 * (r - l) - 2);
    vector<ll> fl(m - l), fr(r - m);
    rep(i, m - l) {
        fl[i] = f[l + i];
    }
    rep(i, r - m) {
        fr[i] = f[m + i];
    }
    {
        auto n12 = convolution_ll(fl, fr);
        rep(i, n12.size()) {
            a12[m - l + i] += n12[i];
        }
        rep(i, l12.size()) {
            a12[i] += l12[i];
        }
        rep(i, r12.size()) {
            a12[2 * (m - l) + i] += r12[i];
        }
    }
    {
        auto n23 = convolution_ll(fl, fr);
        rep(i, n23.size()) {
            a23[m - l + i] += n23[i];
        }
        rep(i, l23.size()) {
            a23[i] += l23[i];
        }
        rep(i, r23.size()) {
            a23[2 * (m - l) + i] += r23[i];
        }
    }
    {
        auto n12_3 = convolution_ll(l12, fr);
        auto n1_23 = convolution_ll(fl, r23);
        rep(i, n12_3.size()) {
            a123[m - l + i] += n12_3[i];
        }
        rep(i, n1_23.size()) {
            a123[2 * (m - l) + i] += n1_23[i];
        }
        rep(i, l123.size()) {
            a123[i] += l123[i];
        }
        rep(i, r123.size()) {
            a123[3 * (m - l) + i] += r123[i];
        }
    }
    return {a12, a23, a123};
}

int main() {
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    int n;
    cin >> n;
    vector<int> is_prime(3 * n + 1, 1);
    is_prime[0] = is_prime[1] = 0;
    for(int i = 2; i <= 3 * n; i++) {
        if(!is_prime[i]) {
            continue;
        }
        for(int j = 2 * i; j <= 3 * n; j += i) {
            is_prime[j] = 0;
        }
    }
    vector<int> f(n + 1);
    rep(i, n + 1) f[i] = is_prime[i];
    auto ret = get<2>(solve(0, n + 1, f));
    ll ans = 0;
    rep(i, 3 * n + 1) {
        ans += is_prime[i] ? ret[i] : 0;
    }
    cout << ans << endl;

    return 0;
}
0