結果

問題 No.8030 ミラー・ラビン素数判定法のテスト
ユーザー kusanokusano
提出日時 2023-01-08 21:27:56
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
(最新)
TLE  
(最初)
実行時間 9,964 ms / 9,973 ms
コード長 1,495 bytes
コンパイル時間 804 ms
コンパイル使用メモリ 64,768 KB
実行使用メモリ 6,820 KB
最終ジャッジ日時 2024-12-16 01:09:23
合計ジャッジ時間 24,753 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,248 KB
testcase_02 AC 2 ms
5,248 KB
testcase_03 AC 2 ms
5,248 KB
testcase_04 AC 5,115 ms
5,248 KB
testcase_05 AC 4,790 ms
5,248 KB
testcase_06 AC 1,210 ms
5,248 KB
testcase_07 AC 1,188 ms
5,248 KB
testcase_08 AC 1,200 ms
5,248 KB
testcase_09 AC 9,964 ms
6,820 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, a_=(a_+a_)%m_)
        if (b_&1)
            r = (r+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