結果

問題 No.3030 ミラー・ラビン素数判定法のテスト
ユーザー kacho65535kacho65535
提出日時 2020-11-15 15:56:18
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
CE  
(最新)
AC  
(最初)
実行時間 -
コード長 2,596 bytes
コンパイル時間 2,236 ms
コンパイル使用メモリ 174,384 KB
最終ジャッジ日時 2024-11-14 23:54:50
合計ジャッジ時間 2,600 ms
ジャッジサーバーID
(参考情報)
judge2 / judge5
このコードへのチャレンジ
(要ログイン)
コンパイルエラー時のメッセージ・ソースコードは、提出者また管理者しか表示できないようにしております。(リジャッジ後のコンパイルエラーは公開されます)
ただし、clay言語の場合は開発者のデバッグのため、公開されます。

コンパイルメッセージ
main.cpp: In instantiation of 'bool primejudge(T) [with T = __int128]':
main.cpp:79:26:   required from here
main.cpp:49:24: error: call of overloaded 'sqrt(__int128&)' is ambiguous
   49 |     double sqrtn = sqrt(n);
      |                    ~~~~^~~
In file included from /usr/include/features.h:486,
                 from /home/linuxbrew/.linuxbrew/Cellar/gcc@12/12.3.0/include/c++/12/x86_64-pc-linux-gnu/bits/os_defines.h:39,
                 from /home/linuxbrew/.linuxbrew/Cellar/gcc@12/12.3.0/include/c++/12/x86_64-pc-linux-gnu/bits/c++config.h:655,
                 from /home/linuxbrew/.linuxbrew/Cellar/gcc@12/12.3.0/include/c++/12/cassert:43,
                 from /home/linuxbrew/.linuxbrew/Cellar/gcc@12/12.3.0/include/c++/12/x86_64-pc-linux-gnu/bits/stdc++.h:33,
                 from main.cpp:4:
/usr/include/x86_64-linux-gnu/bits/mathcalls.h:143:1: note: candidate: 'double sqrt(double)'
  143 | __MATHCALL (sqrt,, (_Mdouble_ __x));
      | ^~~~~~~~~~
In file included from /home/linuxbrew/.linuxbrew/Cellar/gcc@12/12.3.0/include/c++/12/x86_64-pc-linux-gnu/bits/stdc++.h:41:
/home/linuxbrew/.linuxbrew/Cellar/gcc@12/12.3.0/include/c++/12/cmath:467:3: note: candidate: 'constexpr long double std::sqrt(long double)'
  467 |   sqrt(long double __x)
      |   ^~~~
/home/linuxbrew/.linuxbrew/Cellar/gcc@12/12.3.0/include/c++/12/cmath:463:3: note: candidate: 'constexpr float std::sqrt(float)'
  463 |   sqrt(float __x)
      |   ^~~~

ソースコード

diff #

#pragma GCC target("avx2")
#pragma GCC optimize("O3")
#pragma GCC optimize("unroll-loops")
#include <bits/stdc++.h>
using namespace std;
using ull = unsigned long long;
using ll = long long;
#define mod 1000000007ll
#define loop(i, n) for (int i = 0; i < n; i++)
#define flagcount(bit) __builtin_popcount(bit)
#define flag(x) (1ll << x)
#define flagadd(bit, x) bit |= flag(x)
#define flagpop(bit, x) bit &= ~flag(x)
#define flagon(bit, i) bit &flag(i)
#define flagoff(bit, i) !(bit & (1ll << i))
#define all(v) v.begin(), v.end()
#define putout(a) cout << a << '\n'
template <typename T>
bool chmax(T &a, const T &b)
{
    if (a < b)
    {
        a = b; // aをbで更新
        return true;
    }
    return false;
}
template <typename T>
bool chmin(T &a, const T &b)
{
    if (a > b)
    {
        a = b; // aをbで更新
        return true;
    }
    return false;
}
void AS(ll X, ll L, ll R) { assert(L <= X && X <= R); }
using intb=__int128_t;
template <typename T>
bool primejudge(T n)
{
    if (n < 2)
        return false;
    else if (n == 2)
        return true;
    else if (n % 2 == 0)
        return false;
    double sqrtn = sqrt(n);
    for (T i = 3; i < sqrtn + 1; i++)
    {
        if (n % i == 0)
        {
            return false;
        }
        i++;
    }
    return true;
}
//modpow
intb modp(intb a, intb n, intb p)
{
    intb res = 1;
    while (n > 0)
    {
        if (n & 1)
            res = res*a%p;
        a = a*a%p;
        n >>= 1;
    }
    return res;
}

//素数ならtrue,合成数ならfalseを返す
bool MR(intb X)
{

    if (X < 1000)
        return primejudge(X);
    if (X % 2 == 0)
        return false;
    intb Z = X - 1;
    intb s = 0, t = 0;
    while (Z % 2 == 0)
    {
        s++;
        Z /= 2;
    }
    t = Z;
    vector<intb> A = {2, 3, 5, 7, 11, 13, 17, 19, 23, 29, 31, 37};
    for (auto a : A)
    {
        bool ok = true;
        if (modp(a, t, X) == 1)
        {
            ok = false;
            continue;
        }
        intb Z = modp(a, t, X);
        for (ull i = 0; i < s; i++)
        {
            if (Z == X - 1)
            {
                ok = false;
                break;
            }
            Z %= X;
            Z = Z*Z%X;
        }
        if (ok)
        {
            return false; //条件を満たすものが存在しなかった
        }
    }
    return true;
}

int main()
{
    ll Q;
    cin >> Q;
    loop(i, Q)
    {
        ll X;
        cin >> X;
        if (MR(X))
            cout << X << " " << 1 << endl;
        else
            cout << X << " " << 0 << endl;
    }
    return 0;
}
0