結果
問題 | 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 + 7class counting:def __init__(self, n):self.n = nself.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 % modself.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 % moddef comb(self, n, r):if n < r:return 0if n < 0 or r < 0:return 0return self.fa[n] * self.fi[r] % mod * self.fi[n - r] % moddef per(self, n, r):if n < r:return 0if n < 0 or r < 0:return 0return self.fa[n] * self.fi[n - r] % moda, 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 = 0for i in range(n + 1):ans += pow(a * C.comb(n - 1, i) + b * C.comb(n - 1, i - 1), 2, mod)ans %= modprint(ans)