結果

問題 No.577 Prime Powerful Numbers
ユーザー ミドリムシミドリムシ
提出日時 2018-04-21 14:54:05
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 1,943 bytes
コンパイル時間 574 ms
コンパイル使用メモリ 72,380 KB
実行使用メモリ 4,500 KB
最終ジャッジ日時 2023-09-09 12:18:39
合計ジャッジ時間 3,056 ms
ジャッジサーバーID
（参考情報） 		judge15 / judge12
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 7 ms
4,380 KB
testcase_01 WA -
testcase_02 WA -
testcase_03 WA -
testcase_04 WA -
testcase_05 WA -
testcase_06 WA -
testcase_07 WA -
testcase_08 WA -
testcase_09 WA -
testcase_10 AC 2 ms
4,376 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;

bool is_SPRP(uint64_t n, uint64_t a) {
    uint64_t s = __builtin_ctzll(n - 1), d = (n - 1) >> s, cur = 1;
    for (uint64_t pw = d; pw; pw >>= 1) {
    if (pw & 1) cur = ((__uint128_t)cur * a) % n;
        a = ((__uint128_t)a * a) % n;
    }
    if (cur == 1) return true;
    for (; s--; cur = ((__uint128_t)cur * cur) % n)
        if (cur == n - 1) return true;
    return false;
}

bool is_prime(uint64_t x) {
    if (x <= 1) return false;
    vector<uint64_t> bases{2, 325, 9375, 28178, 450775, 9780504, 1795265022};
    if (x < 1ull << 32) bases = {2, 7, 61};

    for (const uint64_t base : bases)
        if (x != base && !is_SPRP(x, base)) return false;
    return true;
}

long power(long base, int exponent){
    long ans = 1;
    for(int i = 0; i < exponent; i++){
        ans *= base;
        if(ans > 6e18 / base){
            return 2e18;
        }
    }
    return ans;
}

bool testcase(long N){
    if(N == 2){
        return false;
    }else if(N % 2 == 0){
        return true;
    }
    for(long i = 2; i < N; i *= 2){
        for(int j = 1; j < 100; j++){
            long left = 1, right = 2e18; // [left, right]
            while(left < right){
                long middle = (left + right) / 2;
                long power_num = power(middle, j);
                if(power_num > N - i){
                    right = middle - 1;    
                }else if(power_num < N - i){
                    left = middle + 1;
                }else if(is_prime(middle)){
                    return true;
                }else{
                    break;
                }
            }
        }
    }
    return false;
}

int main(){
    int Q;
    cin >> Q;
    for(int i = 0; i < Q; i++){
        long N;
        cin >> N;
        if(testcase(N)){
            cout << "Yes" << endl;
        }else{
            cout << "No" << endl;
        }
    }
}
0