結果

問題 No.732 3PrimeCounting
ユーザー iastm
提出日時 2025-10-11 16:22:24
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 946 bytes
コンパイル時間 2,097 ms
コンパイル使用メモリ 135,268 KB
実行使用メモリ 18,068 KB
最終ジャッジ日時 2025-10-11 16:22:34
合計ジャッジ時間 10,288 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 88 WA * 1
権限があれば一括ダウンロードができます

ソースコード

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 + 1, true);
    std::vector<int> primes;
    for (int p = 2; p <= N * 3; p++) {
        if (!is_prime[p]) continue;
        primes.push_back(p);
        if (p >= N * 3 / p) continue;
        for (int k = 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