結果
問題 | No.1693 Invasion |
ユーザー |
![]() |
提出日時 | 2021-10-01 22:33:16 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 97 ms / 2,000 ms |
コード長 | 1,514 bytes |
コンパイル時間 | 160 ms |
コンパイル使用メモリ | 82,696 KB |
実行使用メモリ | 65,664 KB |
最終ジャッジ日時 | 2024-07-19 12:28:57 |
合計ジャッジ時間 | 2,511 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 21 |
ソースコード
mod = 998244353class Combination:"""SIZEが10**6程度以下の二項係数を何回も呼び出したいときに使う使い方:comb = Combination(SIZE, MOD)comb(10, 3) => 120"""def __init__(self, N, MOD=10 ** 9 + 7):self.MOD = MODself.__make_factorial_list(N)def __call__(self, n, k):if k < 0 or k > n:return 0res = self.fact[n] * self.inv[k] % self.MODres = res * self.inv[n - k] % self.MODreturn resdef nPk(self, n, k):if k < 0 or k > n:return 0return self.fact[n] * self.inv[n - k] % self.MODdef nHk(self, n, k):if k == 0:return 1return self.__call__(n + k - 1, k)def __make_factorial_list(self, N):self.fact = [1] * (N + 1)self.inv = [1] * (N + 1)MOD = self.MODfor i in range(1, N + 1):self.fact[i] = (self.fact[i - 1] * i) % MODself.inv[N] = pow(self.fact[N], MOD - 2, MOD)for i in range(N, 0, -1):self.inv[i - 1] = (self.inv[i] * i) % MODreturnN, M = map(int, input().split())A = list(map(int, input().split()))comb = Combination(M + 100, mod)dp = [M + 100] * (M + 1)dp[0] = 0for a in A:for i in range(a, M+1):dp[i] = min(dp[i], dp[i-a] + 1)ans = 0for i in range(M+1):if dp[i] > i:continuebox = M - dp[i]ball = i - dp[i]ans += comb(box, ball)ans %= modprint(ans)