結果

問題 No.3030 ミラー・ラビン素数判定法のテスト
ユーザー kusanokusano
提出日時 2023-01-08 21:31:51
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 3,092 ms / 9,973 ms
コード長 1,615 bytes
コンパイル時間 608 ms
コンパイル使用メモリ 65,152 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-05-09 12:13:15
合計ジャッジ時間 8,784 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 2 ms
5,376 KB
testcase_03 AC 3 ms
5,376 KB
testcase_04 AC 1,609 ms
5,376 KB
testcase_05 AC 1,509 ms
5,376 KB
testcase_06 AC 392 ms
5,376 KB
testcase_07 AC 390 ms
5,376 KB
testcase_08 AC 386 ms
5,376 KB
testcase_09 AC 3,092 ms
5,376 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

long long mulmod(long long a, long long b, long long m)
{
    unsigned long long a_ = (unsigned long long)a;
    unsigned long long b_ = (unsigned long long)b;
    unsigned long long m_ = (unsigned long long)m;
    unsigned long long r = 0;
    for (; b_>0; b_>>=1)
    {
        if (b_&1)
        {
            r += a_;
            if (r>=m_)
                r -= m_;
        }
        a_ += a_;
        if (a_>=m_)
            a_ -= m_;
    }

    return (long long)r;
}

long long powmod(long long a, long long b, long long m)
{
    long long r = 1;
    for (; b>0; b>>=1, a=mulmod(a, a, m))
        if (b&1)
            r = mulmod(r, a, m);
    return r;
}

bool is_prime(long long n)
{
    if (n==1)
        return false;
    if (n==2)
        return true;
    if (n%2==0)
        return false;

    long long d = n-1;
    long long s = 0;
    while (d%2==0)
    {
        d /= 2;
        s++;
    }

    // https://en.wikipedia.org/wiki/Miller%E2%80%93Rabin_primality_test#Testing_against_small_sets_of_bases
    long long A[] = {2, 3, 5, 7, 11, 13, 17, 19, 23, 29, 31, 37};
    for (long long a: A)
    {
        if (a>=n)
            break;
        long long t = powmod(a, d, n);
        bool composite = t!=1;
        for (int i=0; i<s && composite; i++, t=mulmod(t, t, n))
            if (t==n-1)
                composite = false;
        if (composite)
            return false;
    }
    return true;
}

#include <iostream>
using namespace std;

int main()
{
    int n;
    cin>>n;
    for (int i=0; i<n; i++)
    {
        long long x;
        cin>>x;
        cout<<x<<" "<<is_prime(x)<<endl;
    }
}
0