結果

問題 No.3153 probability max K
ユーザー nasutarou1341
提出日時 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
権限があれば一括ダウンロードができます

ソースコード

diff #

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)
0