結果

問題 No.577 Prime Powerful Numbers
コンテスト
ユーザー ミドリムシ
提出日時 2018-04-21 15:45:54
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 2,499 bytes
コンパイル時間 748 ms
コンパイル使用メモリ 76,648 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-06-27 05:23:52
合計ジャッジ時間 1,382 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 3 WA * 7
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <algorithm>
#include <vector>
#include <functional>
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 is_prime_exponentiation(long X){
    for(int i = 1; i < 100; i++){
        long left = 1, right = 2e18; // [left, right]
        while(left <= right){
            long middle = (left + right) / 2;
            long power_num = power(middle, i);
            if(power_num > X){
                right = middle - 1;    
            }else if(power_num < X){
                left = middle + 1;
            }else if(is_prime(middle)){
                return true;
            }else{
                break;
            }
        }
    }
    return false;
}

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 < 2; 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_exponentiation(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