結果

問題 No.109 N! mod M
ユーザー maspymaspy
提出日時 2020-04-02 23:47:34
言語 Python3
(3.12.2 + numpy 1.26.4 + scipy 1.12.0)
結果
AC  
実行時間 900 ms / 5,000 ms
コード長 1,480 bytes
コンパイル時間 158 ms
コンパイル使用メモリ 12,928 KB
実行使用メモリ 48,936 KB
最終ジャッジ日時 2024-06-28 15:26:02
合計ジャッジ時間 6,806 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 524 ms
43,656 KB
testcase_01 AC 803 ms
48,936 KB
testcase_02 AC 900 ms
48,836 KB
testcase_03 AC 515 ms
44,048 KB
testcase_04 AC 520 ms
44,052 KB
testcase_05 AC 864 ms
44,432 KB
testcase_06 AC 524 ms
43,916 KB
testcase_07 AC 554 ms
46,988 KB
testcase_08 AC 514 ms
44,040 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#!/usr/bin/ python3.8
import sys
read = sys.stdin.buffer.read
readline = sys.stdin.buffer.readline
readlines = sys.stdin.buffer.readlines
import numpy as np

T = int(readline())


def MillerRabinTest(n, maxiter=20):
    import random
    if n == 1:
        return False
    elif n == 2:
        return True
    if not n & 1:
        return False
    d = n - 1
    while not (d & 1):
        d >>= 1

    for _ in range(maxiter):
        a = random.randint(1, n - 1)
        t = d
        x = pow(a, t, n)
        while (t != n - 1) and (x != 1) and (x != n - 1):
            x = x * x % n
            t <<= 1
        if (x != n - 1) and not (t & 1):
            return False
    return True


def cumprod(A, MOD):
    L = len(A)
    Lsq = int(L**.5 + 1)
    A = np.resize(A, Lsq**2).reshape(Lsq, Lsq)
    for n in range(1, Lsq):
        A[:, n] *= A[:, n - 1]
        A[:, n] %= MOD
    for n in range(1, Lsq):
        A[n] *= A[n - 1, -1]
        A[n] %= MOD
    return A.ravel()[:L]


def solve():
    N, M = map(int, readline().split())
    if N == 0:
        return 1 % M
    if N >= M:
        return 0
    if M <= 2 * 10 ** 5 + 10:
            x = np.arange(1, N + 1, dtype=np.int64)
            return cumprod(x, M)[-1]
    if not MillerRabinTest(M):
        return 0
    # P is prime
    P = M
    if N == P - 1:
        return P - 1
    x = cumprod(np.arange(P - 1, N, -1, dtype=np.int64), P)[-1]
    return pow(int(-x), P - 2, P)


for _ in range(T):
    print(solve())
0