結果

問題 No.732 3PrimeCounting
コンテスト
ユーザー てんぷら
提出日時 2023-09-22 00:05:58
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 1,823 ms / 3,000 ms
コード長 1,984 bytes
コンパイル時間 3,997 ms
コンパイル使用メモリ 234,188 KB
実行使用メモリ 20,480 KB
最終ジャッジ日時 2024-07-07 17:01:33
合計ジャッジ時間 30,155 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 89
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function 'std::tuple<std::vector<long long int, std::allocator<long long int> >, std::vector<long long int, std::allocator<long long int> > > solve(int, int, std::vector<int>&)':
main.cpp:15:10: warning: structured bindings only available with '-std=c++17' or '-std=gnu++17' [-Wc++17-extensions]
   15 |     auto [l2, l3] = solve(l, m, f);
      |          ^
main.cpp:16:10: warning: structured bindings only available with '-std=c++17' or '-std=gnu++17' [-Wc++17-extensions]
   16 |     auto [r2, r3] = solve(m, r, f);
      |          ^

ソースコード

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>> solve(int l, int r, vector<int> &f) {
    if(r - l == 1) {
        return {{0}, {0}};
    }
    int m = (l + r) / 2;
    auto [l2, l3] = solve(l, m, f);
    auto [r2, r3] = solve(m, r, f);
    vector<ll> a2(2 * (r - l) - 1), a3(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 n2 = convolution_ll(fl, fr);
        rep(i, n2.size()) {
            a2[m - l + i] += n2[i];
        }
        rep(i, l2.size()) {
            a2[i] += l2[i];
        }
        rep(i, r2.size()) {
            a2[2 * (m - l) + i] += r2[i];
        }
    }
    {
        auto n2_1 = convolution_ll(l2, fr);
        auto n1_2 = convolution_ll(fl, r2);
        rep(i, n2_1.size()) {
            a3[m - l + i] += n2_1[i];
        }
        rep(i, n1_2.size()) {
            a3[2 * (m - l) + i] += n1_2[i];
        }
        rep(i, l3.size()) {
            a3[i] += l3[i];
        }
        rep(i, r3.size()) {
            a3[3 * (m - l) + i] += r3[i];
        }
    }
    return {a2, a3};
}

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<1>(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