結果
問題 | No.2792 Security Cameras on Young Diagram |
ユーザー |
|
提出日時 | 2024-06-21 21:49:28 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 236 ms / 2,000 ms |
コード長 | 1,449 bytes |
コンパイル時間 | 329 ms |
コンパイル使用メモリ | 82,304 KB |
実行使用メモリ | 114,816 KB |
最終ジャッジ日時 | 2024-06-24 18:44:46 |
合計ジャッジ時間 | 6,426 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 21 |
ソースコード
# import pypyjit# pypyjit.set_param('max_unroll_recursion=-1')from collections import defaultdict as ddS = inputR = rangeP = printdef I(): return int(S())def M(): return map(int, S().split())def L(): return list(M())def O(): return list(map(int, open(0).read().split()))def yn(b): print("Yes" if b else "No")biga = "ABCDEFGHIJKLMNOPQRSTUVWXYZ"smaa = "abcdefghijklmnopqrstuvwxyz"def acc(a):b = [0]for i in a:b.append(b[-1] + i)return bclass Comb:def __init__(self, N, mod = 1000000007):self.n = Nself.mod = modself.fac = [0] * (self.n + 1)self.invf = [0] * (self.n + 1)self.inv = [0] * (self.n + 1)self.fac[0] = 1self.fac[1] = 1self.invf[0] = 1self.invf[1] = 1self.inv[1] = 1for i in range(2, self.n + 1):self.fac[i] = self.fac[i - 1] * i % self.modself.inv[i] = self.mod - self.inv[self.mod % i] * (self.mod // i) % self.modself.invf[i] = self.invf[i - 1] * self.inv[i] % self.moddef F(self, N):return self.fac[N]def C(self, N, K):return self.invf[K] * self.invf[N - K] % self.mod * self.fac[N] % self.moddef P(self, N, K):return self.invf[N - K] * self.fac[N] % self.moddef H(self, N, K):return self.invf[K] * self.invf[N - 1] % self.mod * self.fac[N + K - 1] % self.modN = int(input())A = L() + [-1]c = Comb(10 ** 6, 998244353)mod = 998244353ans = 0for i in range(N):for j in range(A[i + 1] + 1, A[i] + 1):ans += c.C(i + j, i)print(ans % mod)