結果
問題 |
No.3153 probability max K
|
ユーザー |
![]() |
提出日時 | 2025-05-20 21:31:43 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 285 ms / 2,000 ms |
コード長 | 859 bytes |
コンパイル時間 | 413 ms |
コンパイル使用メモリ | 82,376 KB |
実行使用メモリ | 102,400 KB |
最終ジャッジ日時 | 2025-05-20 21:31:58 |
合計ジャッジ時間 | 7,265 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 20 |
ソースコード
class Inv: def __init__(s, mod): s.MOD = mod def modpow(s, a, n): res = 1 while n > 0: if n & 1: res = res * a % s.MOD a = a * a % s.MOD n >>= 1 return res def invx(s, a): return s.modpow(a, s.MOD - 2) def invpowL(s, a, n): # a^-x (0 <= x <= n)リスト ia = s.invx(a) L = [1] * (n + 1) for i in range(1, n + 1): L[i] = L[i - 1] * ia % s.MOD return L def invL(s, n): # 0 <= x <= n 逆元リスト I = [0, 1] for i in range(2, n + 1): I.append(s.MOD - I[s.MOD % i] * (s.MOD // i) % s.MOD) return I N, K = list(map(int, input().split())) A = list(map(int, input().split())) MOD = 998244353 inv = Inv(MOD) k = 1 kk = 1 for a in A: n = inv.invx(a) k *= min(a, K) * n % MOD kk *= min(a, (K - 1)) * n % MOD k %= MOD kk %= MOD ans = k - kk print(ans % MOD)