結果

問題 No.843 Triple Primes
コンテスト
ユーザー hipopo
提出日時 2020-01-21 00:42:03
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 35 ms / 2,000 ms
コード長 1,143 bytes
コンパイル時間 1,296 ms
コンパイル使用メモリ 110,808 KB
最終ジャッジ日時 2025-01-08 20:22:18
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 42
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include <algorithm>
#include <cmath>
#include <complex>
#include <iostream>
#include <map>
#include <queue>
#include <set>
#include <vector>

using namespace std;

using ll = long long;
using P = pair<int, int>;

const long long MOD = 1e9+7;

template<class T> inline bool chmax(T& a, T b) { if (a < b) { a = b; return 1; } return 0; }
template<class T> inline bool chmin(T& a, T b) { if (a > b) { a = b; return 1; } return 0; }

vector<int> sieve(int n) {
    vector<int> is_prime(n + 1, true);
    for (int i = 2; i <= n; i++) {
        if (!is_prime.at(i)) continue;
        
        is_prime.at(i) = true;
        for (int j = i + i; j <= n; j += i) is_prime.at(j) = false;
    }
    vector<int> res;
    for (int i = 2; i <= n; i++) if (is_prime.at(i)) res.push_back(i);
    return res;
}

int main() { 
    int n;
    cin >> n;
    vector<int> primes = sieve(n);
    map<ll, int> is_square_prime;
    for (int p: primes) {
        is_square_prime[(ll)p * p]++;
    }

    int cnt = 0;
    for (int p: primes) {
        cnt += is_square_prime[2 + p];
        if (p != 2) cnt += is_square_prime[2 + p];
    }
    cout << cnt << endl;
}
0