結果
問題 | No.2792 Security Cameras on Young Diagram |
ユーザー |
![]() |
提出日時 | 2024-06-21 22:03:11 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 378 ms / 2,000 ms |
コード長 | 585 bytes |
コンパイル時間 | 181 ms |
コンパイル使用メモリ | 82,176 KB |
実行使用メモリ | 240,176 KB |
最終ジャッジ日時 | 2024-06-24 18:45:05 |
合計ジャッジ時間 | 9,826 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 21 |
ソースコード
mod = 998244353n = 10**7fact = [1 for i in range(n+1)]fact_inv = [1 for i in range(n+1)]for i in range(1,n+1):fact[i] = fact[i-1]*i % modfact_inv[-1] = pow(fact[-1],mod-2,mod)for i in range(n,0,-1):fact_inv[i-1] = fact_inv[i]*i % moddef binom(n,r):if n < r or n < 0 or r < 0:return 0res = fact_inv[n-r] * fact_inv[r] % modres *= fact[n]res %= modreturn resN = int(input())A = list(map(int,input().split()))ans = 0for i in range(1,N):a,aa = A[i-1],A[i]ans += binom(a+i,i) - binom(aa+i,i)ans %= modans += binom(A[-1]+N,N)print(ans % mod)