結果

問題 No.109 N! mod M
ユーザー colognecologne
提出日時 2022-02-02 17:12:30
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 174 ms / 5,000 ms
コード長 884 bytes
コンパイル時間 203 ms
コンパイル使用メモリ 82,560 KB
実行使用メモリ 75,264 KB
最終ジャッジ日時 2024-06-11 09:29:25
合計ジャッジ時間 1,334 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 38 ms
57,216 KB
testcase_01 AC 122 ms
57,856 KB
testcase_02 AC 146 ms
57,856 KB
testcase_03 AC 36 ms
51,968 KB
testcase_04 AC 42 ms
59,392 KB
testcase_05 AC 174 ms
75,264 KB
testcase_06 AC 68 ms
75,008 KB
testcase_07 AC 44 ms
57,344 KB
testcase_08 AC 39 ms
58,112 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import itertools
import sys


def input(): return sys.stdin.readline().rstrip()


def isPrime(N):
    for i in itertools.count(2):
        if i*i > N:
            return True
        if N % i == 0:
            return False


def fact(N, M):
    # M - N <= LIM
    LIM = 10**5

    if N >= M:
        # trivial case
        return 0
    if M <= 2*LIM:
        # small case
        a = 1 % M
        for i in range(1, N+1):
            a = a * i % M
        return a
    elif isPrime(M):
        # prime case, (M-1)! == -1 (mod M)
        a = M-1
        for i in range(N+1, M):
            a = a * i % M
        return pow(a, -1, M)
    else:
        # M = pq, q <= M/p <= M/2 <= M/2 + (M/2 - LIM) <= N
        return 0


def main():
    T = int(input())
    for i in range(T):
        N, M = map(int, input().split())
        print(fact(N, M))


if __name__ == '__main__':
    main()
0