結果

問題 No.577 Prime Powerful Numbers
ユーザー xuzijian629xuzijian629
提出日時 2018-11-01 23:03:57
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
TLE  
実行時間 -
コード長 2,272 bytes
コンパイル時間 1,946 ms
コンパイル使用メモリ 203,808 KB
実行使用メモリ 17,080 KB
最終ジャッジ日時 2024-04-30 12:04:35
合計ジャッジ時間 8,423 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

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

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
using i64 = int64_t;
using vi = vector<i64>;
using vvi = vector<vi>;
mt19937 rnd(chrono::steady_clock::now().time_since_epoch().count());

i64 modpow(i64 a, i64 n, i64 mod) {
    if (n == 0) return 1;
    if (n % 2 == 0) {
        i64 t = modpow(a, n / 2, mod);
        return t * t % mod;
    }
    return a * modpow(a, n - 1, mod) % mod;
}

bool is_prime(i64 n, int k = 10) {
    if (n == 2) return true;
    if (n < 2 || n % 2 == 0) return false;
    i64 d = n - 1;
    while (d % 2 == 0) {
        d /= 2;
    }
    for (int i = 0; i < k; i++) {
        i64 a = rnd() % (n - 2) + 1;
        i64 t = d;
        i64 y = modpow(a, t, n);
        while (t != n - 1 && y != 1 && y != n - 1) {
            y = modpow(y, 2, n);
            t *= 2;
        }
        if (y != n - 1 && t % 2 == 0) {
            return false;
        }
    }
    return true;
}

__int128_t powi(i64 n, int k) {
    __int128_t ret = 1;
    for (int i = 0; i < k; i++) {
        ret *= n;
    }
    return ret;
}

i64 ksqrt(i64 n, int k) {
    i64 l = 1, r = n;
    while (l < r - 1) {
        i64 m = (l + r) / 2;
        if (powi(m, k) <= n) {
            l = m;
        } else {
            r = m;
        }
    }
    return l;
}

int main() {
    int q;
    cin >> q;
    while (q--) {
        i64 n;
        cin >> n;
        if (n <= 2) {
            cout << "No" << endl;
        } else if (n % 2 == 0) {
            cout << "Yes" << endl;
        } else {
            int ok = 0;
            for (i64 i = 2; i < n; i <<= 1) {
                if (ok) break;
                i64 k = n - i;
                if (is_prime(k)) {
                    cout << "Yes" << endl;
                    ok = 1;
                    break;
                }
                if (!ok) {
                    for (int i = 2;; i++) {
                        i64 s = ksqrt(k, i);
                        if (s == 1) break;
                        if (powi(s, i) == k) {
                            cout << "Yes" << endl;
                            ok = 1;
                            break;
                        }
                    }   
                }
            }
            if (!ok) {
                cout << "No" << endl;
            }
        }
    }
}
0