結果

問題 No.2792 Security Cameras on Young Diagram
ユーザー ButterflvButterflv
提出日時 2024-06-21 22:38:54
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 151 ms / 2,000 ms
コード長 481 bytes
コンパイル時間 164 ms
コンパイル使用メモリ 82,588 KB
実行使用メモリ 105,104 KB
最終ジャッジ日時 2024-06-24 18:45:59
合計ジャッジ時間 3,793 ms
ジャッジサーバーID
(参考情報)
judge3 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 21
権限があれば一括ダウンロードができます

ソースコード

diff #

N=int(input())
A=tuple(map(int,input().split()))
MOD=998244353
inv=[None]+[1]+[None]*(3*10**5+100)

for a in range(2, len(inv)):
  inv[a] = -(MOD // a)*inv[MOD % a]
  inv[a] %= MOD

fac=[1, 1]+[None]*(3*10**5)
for i in range(2, len(fac)):
  fac[i] = (fac[i-1]*i)%MOD

facInv=[1, 1]+[None]*(3*10**5)
for i in range(2, len(facInv)):
  facInv[i] = (facInv[i-1]*inv[i])%MOD

ans = 1

for i in range(len(A)):
  ans += fac[A[i]-1+i+1]*facInv[A[i]-1]*facInv[i+1]
  ans %= MOD

print(ans)
0