結果
問題 |
No.1127 変形パスカルの三角形
|
ユーザー |
|
提出日時 | 2020-11-21 14:29:09 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 523 ms / 1,500 ms |
コード長 | 952 bytes |
コンパイル時間 | 151 ms |
コンパイル使用メモリ | 12,800 KB |
実行使用メモリ | 26,496 KB |
最終ジャッジ日時 | 2024-07-23 15:37:50 |
合計ジャッジ時間 | 10,798 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 30 |
ソースコード
mod = 10 ** 9 + 7 class counting: def __init__(self, n): self.n = n self.fa = [1] * (self.n + 1) self.fi = [1] * (self.n + 1) for i in range(1, self.n + 1): self.fa[i] = self.fa[i - 1] * i % mod self.fi[-1] = pow(self.fa[-1], mod - 2, mod) for i in range(self.n, 0, -1): self.fi[i - 1] = self.fi[i] * i % mod def comb(self, n, r): if n < r:return 0 if n < 0 or r < 0:return 0 return self.fa[n] * self.fi[r] % mod * self.fi[n - r] % mod def per(self, n, r): if n < r:return 0 if n < 0 or r < 0:return 0 return self.fa[n] * self.fi[n - r] % mod a, b = map(int, input().split()) n, k = map(int, input().split()) C = counting(n) print((a * C.comb(n - 1, k - 1) + b * C.comb(n - 1, k - 2)) % mod) ans = 0 for i in range(n + 1): ans += pow(a * C.comb(n - 1, i) + b * C.comb(n - 1, i - 1), 2, mod) ans %= mod print(ans)