結果

問題 No.843 Triple Primes
コンテスト
ユーザー gon_027
提出日時 2020-02-11 00:36:25
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 40 ms / 2,000 ms
コード長 1,156 bytes
コンパイル時間 759 ms
コンパイル使用メモリ 84,792 KB
実行使用メモリ 6,820 KB
最終ジャッジ日時 2024-10-01 07:12:31
合計ジャッジ時間 2,463 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 42
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <cmath>
#include <string>
#include <cstdio>
#include <algorithm>
#include <map>
#include <queue>
using namespace std;

#define rep(i, n) for(int (i) = (0); (i) < (n); (i++))
using ll = long long;

template <class T>
bool isPrime(T num){
    if(num <= 1) return false;
    if(num == 2) return true;
    if(num % 2 == 0) return false;

    for(int i = 3; i * i <= num; i += 2){
        if(num % i == 0){
            return false;
        }
    }

    return true;
}

template <class T>
void primeVector(vector<T>& v, T start, T end){
    for(int i = start; i <= end; ++i){
        if(isPrime(i)){
            v.push_back(i);
        }
    }
}

vector<int> v;
bool bp[500010];

int main() {
    int n;
    cin >> n;
    primeVector(v, 0, n);
    for(int i = 0; i < v.size(); ++i){
        bp[v[i]] = true;
    }

    if(n == 1){
        cout << 0 << endl;
        return 0;
    }

    int ans = 1;
    for(int i = 1; i < v.size() && v[i] <= n; ++i){
        int r = 2 + v[i];
        int rr = floor(sqrt(r));

        if(rr * rr == r && bp[rr]) {
            ans += 2;
        }
    }

    cout << ans << endl;
}
0