結果

問題 No.732 3PrimeCounting
コンテスト
ユーザー iastm
提出日時 2025-10-11 16:38:29
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 360 ms / 3,000 ms
コード長 916 bytes
コンパイル時間 2,147 ms
コンパイル使用メモリ 135,136 KB
実行使用メモリ 18,004 KB
最終ジャッジ日時 2025-10-11 16:38:39
合計ジャッジ時間 9,739 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 89
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <atcoder/convolution>

using ll = long long;

int main() {
    int N;
    std::cin >> N;
    std::vector<bool> is_prime(N * 3, true);
    std::vector<int> primes;
    for (int p = 2; p < N * 3; p++) {
        if (!is_prime[p]) continue;
        if (p > 2) primes.push_back(p);
        for (ll k = ll(p) * p; k < N * 3; k += p) is_prime[k] = false;
    }
    std::vector<ll> fps(N + 1);
    for (int p : primes) {
        if (p <= N) fps[p] = 1;
        else break;
    }
    std::vector<ll> comp = atcoder::convolution_ll(fps, fps);
    comp = atcoder::convolution_ll(comp, fps);
    for (int p : primes) {
        if (p > N) break;
        for (int q : primes) {
            if (q > N) break;
            comp[p + p + q] -= 3;
        }
    }
    int64_t result = 0;
    for (int p : primes) result += comp[p];
    result /= 6;
    std::cout << result << std::endl;
}
0