結果

問題 No.2795 Perfect Number
コンテスト
ユーザー あ
提出日時 2024-06-28 23:19:25
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 2 ms / 2,000 ms
コード長 1,305 bytes
コンパイル時間 456 ms
コンパイル使用メモリ 65,592 KB
実行使用メモリ 6,948 KB
最終ジャッジ日時 2024-06-28 23:19:28
合計ジャッジ時間 1,363 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 35
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <cstdint>
using namespace std;
using ull = unsigned long long;

__int128_t pow(__int128_t a, __int128_t n, __int128_t m) {
    __int128_t x = a;
    __int128_t ret = 1ull;
    while (n) {
        if (n & 1ull) {
            ret *= x;
            ret %= m;
        }
        x *= x;
        x %= m;
        n = (n >> 1);
    }
    return ret;
}

bool MR(ull n, int k) {
    if (n == 2) return true;
    if (n%2==0) return false;
    if (n == 1) return false;
    int s = 1;
    while ((((n-1)>>s) & 1ull) == 0ull) s++;
    ull t = (n-1)>>s;
    for (int i=0;i<k;i++) {
        ull a = (ull)((double)rand()/RAND_MAX*(n-2))+1;
        __int128_t a_sq = pow(a, t, n);
        bool f = false;
        if (a_sq == 1) continue;
        for (int j = 0;j<s;j++) {
            if (a_sq == n-1) {
                f = true;
                break;
            }
            a_sq *= a_sq;
            a_sq %= n;
        }
        if (!f) return false;
    }
    return true;
}

int main( ){
    ull x;
    cin >> x;

    int c = 0;
    while (!((x >> c) & 1ull)) {
        c++;
    }
    if ((x >> c) == ((1ull << (c+1))-1)) {
        if (MR((1ull<<(c+1))-1, 10)) {
            cout << "Yes";
        }  else {
            cout << "No";
        }
    } else {
        cout << "No";
    }
}
0