結果

問題 No.732 3PrimeCounting
コンテスト
ユーザー merom686
提出日時 2018-09-11 20:19:48
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 128 ms / 3,000 ms
コード長 1,096 bytes
コンパイル時間 865 ms
コンパイル使用メモリ 80,732 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-06-23 22:20:54
合計ジャッジ時間 5,122 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 89
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <string>
#include <algorithm>
#include <cstdio>
#include <cstring>
#include <cmath>
using namespace std;
using ll = long long;

vector<int> prime_vector(int n) {
    int m = (int)sqrt(n - 1);

    vector<bool> q(n);
    for (int i = 2; i <= m; i++) {
        if (q[i]) continue;
        for (int j = i * i; j < n; j += i) {
            q[j] = true;
        }
    }

    vector<int> p; p.reserve(n / log(n * 0.329));
    for (int i = 2; i < n; i++) {
        if (!q[i]) p.push_back(i);
    }
    return p;
}

int main() {
    int n;
    cin >> n;

    auto pv = prime_vector(n * 3);
    int m = pv.size();

    vector<int> s(n * 2);
    ll r = 0;

    for (int i = 3; i < m; i++) {
        int c = pv[i];
        if (c > n) break;

        int b = pv[i - 1];
        for (int j = 1; j < i - 1; j++) {
            int a = pv[j];
            s[a + b]++;
        }

        for (int j = i + 1; j < m; j++) {
            int t = pv[j] - c;
            if (t >= b * 2) break;
            r += s[t];
        }
    }

    cout << r << endl;

    return 0;
}
0