結果
問題 | No.2679 MODice |
ユーザー | Ekiben542 |
提出日時 | 2024-03-20 21:12:46 |
言語 | PyPy3 (7.3.15) |
結果 |
RE
|
実行時間 | - |
コード長 | 801 bytes |
コンパイル時間 | 153 ms |
コンパイル使用メモリ | 81,700 KB |
実行使用メモリ | 847,280 KB |
最終ジャッジ日時 | 2024-03-20 21:12:53 |
合計ジャッジ時間 | 2,794 ms |
ジャッジサーバーID (参考情報) |
judge11 / judge12 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 37 ms
53,460 KB |
testcase_01 | AC | 37 ms
53,460 KB |
testcase_02 | AC | 36 ms
53,460 KB |
testcase_03 | AC | 37 ms
53,460 KB |
testcase_04 | AC | 37 ms
53,460 KB |
testcase_05 | AC | 37 ms
53,460 KB |
testcase_06 | RE | - |
testcase_07 | RE | - |
testcase_08 | RE | - |
testcase_09 | MLE | - |
testcase_10 | -- | - |
testcase_11 | -- | - |
testcase_12 | -- | - |
testcase_13 | -- | - |
testcase_14 | -- | - |
testcase_15 | -- | - |
ソースコード
def dice_remainder_probability(N, K): dp = [0 for _ in range(6*N+1)] dp[0] = 1 # DP for i in range(N): dp_new = [0 for _ in range(6*(i+1)+1)] for j in range(6*i+1): for k in range(1, 7): dp_new[j+k] += dp[j] dp = dp_new total = 0 for i in range(K, 6*N+1, 6): total += dp[i] return total, 6**N def mod_inverse(a, m): g, x, y = extended_gcd(a, m) if g != 1: raise Exception('M') else: return x % m def extended_gcd(a, b): if a == 0: return b, 0, 1 else: g, x, y = extended_gcd(b % a, a) return g, y - (b // a) * x, x N, K = map(int, input().split()) P, Q = dice_remainder_probability(N, K) R = (P * mod_inverse(Q, 998244353)) % 998244353 print(R)