結果

問題 No.3030 ミラー・ラビン素数判定法のテスト
ユーザー AC2KAC2K
提出日時 2023-03-31 15:05:14
言語 C++23
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 253 ms / 9,973 ms
コード長 2,155 bytes
コンパイル時間 2,865 ms
コンパイル使用メモリ 244,784 KB
実行使用メモリ 6,944 KB
最終ジャッジ日時 2024-09-22 19:11:20
合計ジャッジ時間 3,938 ms
ジャッジサーバーID
(参考情報)
judge5 / judge3
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,816 KB
testcase_01 AC 1 ms
6,944 KB
testcase_02 AC 2 ms
6,940 KB
testcase_03 AC 2 ms
6,944 KB
testcase_04 AC 141 ms
6,940 KB
testcase_05 AC 137 ms
6,940 KB
testcase_06 AC 57 ms
6,940 KB
testcase_07 AC 56 ms
6,940 KB
testcase_08 AC 57 ms
6,944 KB
testcase_09 AC 253 ms
6,940 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

#include<bits/stdc++.h>
template <class T, class U = T>
constexpr inline U mod_pow(T base, T exp, T mod) {
    T ans = 1;
    base %= mod;
    while (exp > 0) {
        if (exp & 1) {
            ans *= base;
            ans %= mod;
        }
        base *= base;
        base %= mod;
        exp >>= 1;
    }
    return ans;
}

namespace prime {
    namespace miller {
        using i128 = __int128_t;
        using u128 = __uint128_t;
        using u64 = uint64_t;
        using u32 = uint32_t;

        bool miller_rabin(u64 n, const u64 bases[], int length) {
            u64 d = n - 1;

            while (~d & 1) {
                d >>= 1;
            }

            u64 rev = n - 1;

            for (int i = 0; i < length; i++) {
                u64 a = bases[i];

                if (n <= a) {
                    return true;
                }
                u64 t = d;
                u128 y = mod_pow<u128>(a, t, n);
                while (t != n - 1 && y != 1 && y != rev) {
                    (y *= y) %= n;
                    t <<= 1;
                }

                if (y != rev && (~t & 1))return false;
            }
            return true;
        }


        constexpr u64 bases_int[3] = { 2, 7, 61 };  // intだと、2,7,61で十分
        constexpr u64 bases_ll[7] = { 2, 325, 9375, 28178, 450775, 9780504, 1795265022 };
        bool is_prime(u64 n) {
            if (n < 2) {
                return false;
            }
            else if (n == 2) {
                return true;
            }
            else if (~n & 1) {
                return false;
            }
            if (n < (1ul << 31)) {
                return miller_rabin(n, bases_int, 3);
            }
            else {
                return miller_rabin(n, bases_ll, 7);
            }
        }
    };
};
///@brief fast prime check(MillerRabinの素数判定)


int main() {
    int n;
    scanf("%d", &n);
    for (int i = 0; i < n; i++) {
        uint64_t xi;
        scanf("%lld", &xi);
        printf("%lld ", xi);
        if (prime::miller::is_prime(xi)) {
            puts("1");
        }
        else {
            puts("0");
        }
    }
}
0