結果

問題 No.732 3PrimeCounting
ユーザー kokatsu
提出日時 2022-01-02 22:46:10
言語 D
(dmd 2.109.1)
結果
AC  
実行時間 645 ms / 3,000 ms
コード長 859 bytes
コンパイル時間 1,778 ms
コンパイル使用メモリ 208,744 KB
実行使用メモリ 6,948 KB
最終ジャッジ日時 2024-06-22 13:58:21
合計ジャッジ時間 9,543 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 89
権限があれば一括ダウンロードができます

ソースコード

diff # 

import std;

void main() {
    int N;
    readf("%d\n", N);

    int M = N * 3;
    int S = M.to!real.sqrt.floor.to!int;
    auto sieve = iota(0, M+1).map!(i => i == 2 || (i > 2 && i % 2 == 1)).array;
    foreach (i; iota(3, S+1, 2)) {
        if (!sieve[i]) {
            continue;
        }

        foreach (j; iota(i*i, M+1, i)) {
            sieve[j] = false;
        }
    }

    auto primes = iota(0, M+1).filter!(i => sieve[i]).array;
    auto len = primes.length;

    long res;
    auto cnts = new long[](M);

    foreach (i; 2 .. len) {
        int b = primes[i-1], c = primes[i];

        if (c > N) break;

        foreach (j; 0 .. i-1) {
            int a = primes[j];

            ++cnts[a+b];
        }

        foreach (p; primes) {
            if (p >= c) {
                res += cnts[p-c];
            }
        }
    }

    res.writeln;
}
0