結果
問題 | No.2792 Security Cameras on Young Diagram |
ユーザー |
|
提出日時 | 2024-06-21 22:15:42 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 241 ms / 2,000 ms |
コード長 | 1,136 bytes |
コンパイル時間 | 248 ms |
コンパイル使用メモリ | 82,304 KB |
実行使用メモリ | 112,896 KB |
最終ジャッジ日時 | 2024-06-24 18:45:38 |
合計ジャッジ時間 | 6,276 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 21 |
ソースコード
class Combination:def __init__(self, MX=10**6, MOD=998244353):self.MX = MXself.MOD = MODself.fact = [1] * (MX + 1)self.inv = [1] * (MX + 1)self.f_inv = [1] * (MX + 1)for i in range(2, MX + 1):self.fact[i] = (self.fact[i - 1] * i) % self.MODself.inv[i] = ( - (self.MOD // i) * (self.inv[self.MOD % i])) % self.MODself.f_inv[i] = (self.f_inv[i - 1] * self.inv[i]) % self.MODdef invs(self, n):if n <= self.MX:return self.inv[n]else:return pow(n, self.MOD - 2, self.MOD)def p(self, n, r):if r > n or r < 0:return 0return (self.fact[n] * self.f_inv[r]) % self.MODdef c(self, n, r):if r > n or r < 0:return 0return (self.fact[n] * self.f_inv[r] * self.f_inv[n - r]) % self.MODMOD = 998244353com = Combination()N = int(input())A = list(map(int, input().split()))ans = 1ans %= MODfor i in range(N):ans += com.c(i + A[i], i + 1)ans %= MOD# ans += com.c(N + A[-1], N)ans %= MODprint(ans)