結果

問題 No.843 Triple Primes
ユーザー ManjushriMitra
提出日時 2024-07-27 12:09:31
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 15 ms / 2,000 ms
コード長 1,169 bytes
コンパイル時間 1,936 ms
コンパイル使用メモリ 171,020 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-07-27 12:09:35
合計ジャッジ時間 4,147 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 42
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
using namespace std;
using ll = long long;
#define rep(i,m,n) for(int i=m; i<n; ++i)
#define repl(i,m,n) for(ll i=m; i<n; ++i)

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

    vector<bool> is_prime(500010, true);
    vector<int> prime_list;

    is_prime[0] = is_prime[1] = false;
    for(int i=2; i<=500010; ++i){
        if(!is_prime[i]) continue;
        prime_list.push_back(i);
        for(int j=2*i; j<=500010; j+=i){
            is_prime[j] = false;
        }
    }

    int ans = 0;
    if(N <= 100){
        for(int i=0; prime_list[i] <= N; ++i){
            for(int j=0; prime_list[j] <= N; ++j){
                for(int k=0; prime_list[k] <= N; ++k){
                    int p = prime_list[i], q = prime_list[j], r = prime_list[k];
                    if(p + q == r*r) ans++;
                }
            }
        }
    }else{
        for(int r : prime_list){
            if(r*r >= 2*N || r > N) break;
            for(int p : prime_list){
                if(r*r <= p || p > N) break;
                int q = r*r - p;
                if(is_prime[q] && q <= N) ans++;
            }
        }
    }
    cout << ans << endl;

    return 0;
}
0