結果

問題 No.843 Triple Primes
コンテスト
ユーザー kibuna
提出日時 2019-06-28 21:46:49
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 109 ms / 2,000 ms
コード長 1,729 bytes
コンパイル時間 1,678 ms
コンパイル使用メモリ 173,260 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-09-19 13:59:22
合計ジャッジ時間 5,067 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 42
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include "bits/stdc++.h"
using namespace std;
using ll = long long;
using pii = pair<int, int>;
using pll = pair<ll, ll>;
using vi = vector<int>;
using vl = vector<ll>;
using vvi = vector<vi>;
using vvl = vector<vl>;
const int INF = 1 << 28;
const ll MOD = 1000000007;
template <class T>
bool chmax(T &a, const T &b) {
    return (a < b) ? (a = b, 1) : 0;
}
template <class T>
bool chmin(T &a, const T &b) {
    return (b < a) ? (a = b, 1) : 0;
}

struct Sieve {
    vector<bool> isPrime;
    Sieve(size_t max) : isPrime(max + 1, true) {
        isPrime[0] = false; // 0は素数ではない
        isPrime[1] = false; // 1は素数ではない

        for (size_t i = 2; i * i <= max; ++i)         // 0からsqrt(max)まで調べる
            if (isPrime[i])                           // iが素数ならば
                for (size_t j = 2; i * j <= max; ++j) // (max以下の)iの倍数は
                    isPrime[i * j] = false;           // 素数ではない
    }
    bool operator()(size_t n) { return isPrime[n]; }
};

int main() {
    int n;
    cin >> n;
    int m = int(sqrt(double(2 * n)));
    Sieve si(n);
    vi primesr;
    vi primespq;
    primesr.reserve(n);
    primespq.reserve(n);
    for (int i = 2; i <= m; ++i) {
        if (si(i))
            primesr.push_back(i);
    }
    for (int i = 2; i <= n; ++i) {
        if (si(i))
            primespq.push_back(i);
    }
    int ret = 0;
    for (auto &r : primesr) {
        for (auto &p : primespq) {
            if (p > r * r / 2)
                break;
            if (binary_search(primespq.begin(), primespq.end(), r * r - p))
                ret += 2;
        }
    }
    if (n > 1)
        ret -= 1;
    cout << ret << "\n";

    return 0;
}
0