結果
問題 | No.1785 Inequality Signs |
ユーザー |
|
提出日時 | 2022-10-26 14:49:40 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 358 ms / 2,000 ms |
コード長 | 1,212 bytes |
コンパイル時間 | 372 ms |
コンパイル使用メモリ | 82,488 KB |
実行使用メモリ | 101,748 KB |
最終ジャッジ日時 | 2024-07-04 04:38:08 |
合計ジャッジ時間 | 14,138 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 52 |
ソースコード
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) % MODself.inv[i] = ( - (MOD // i) * (self.inv[MOD % i])) % MODself.f_inv[i] = (self.f_inv[i - 1] * self.inv[i]) % 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 = 10 ** 9 + 7com = Combination(MOD = MOD)N, K = map(int, input().split())ans = 0tmp = 1for j in range(N):tmp *= K + jtmp %= MODfor i in range(N):ans += tmp * com.c(N - 1, i)ans %= MODtmp *= K - i - 1tmp *= com.invs(K + N - 1 - i)tmp %= MODans *= com.f_inv[N]ans %= MODprint(ans)