結果
問題 |
No.1127 変形パスカルの三角形
|
ユーザー |
![]() |
提出日時 | 2020-08-27 18:09:37 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 369 ms / 1,500 ms |
コード長 | 1,058 bytes |
コンパイル時間 | 222 ms |
コンパイル使用メモリ | 82,432 KB |
実行使用メモリ | 80,640 KB |
最終ジャッジ日時 | 2024-11-07 22:34:33 |
合計ジャッジ時間 | 8,895 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 30 |
ソースコード
import sys; input = sys.stdin.buffer.readline sys.setrecursionlimit(10**7) from collections import defaultdict con = 10 ** 9 + 7; INF = float("inf") def getlist(): return list(map(int, input().split())) class Combination(object): def __init__(self, N, con): self.fac = [0] * (N + 1) self.inv = [0] * (N + 1) self.finv = [0] * (N + 1) self.fac[0], self.fac[1] = 1, 1 self.inv[1] = 1 self.finv[0], self.finv[1] = 1, 1 for i in range(2, N + 1): self.fac[i] = self.fac[i - 1] * i % con self.inv[i] = self.inv[con % i] * (con - (con // i)) % con self.finv[i] = self.finv[i - 1] * self.inv[i] % con def com(self, N, k): return (self.fac[N] * self.finv[k] * self.finv[N - k]) % con #処理内容 def main(): a, b = getlist() N, K = getlist() C = Combination(N - 1, con) ans1 = a * C.com(N - 1, K - 1) + b * C.com(N - 1, K - 2) ans1 %= con print(ans1) ans2 = a ** 2 + b ** 2 for i in range(N - 1): ans2 += (a * C.com(N - 1, i + 1) + b * C.com(N - 1, i)) ** 2 ans2 %= con print(ans2) if __name__ == '__main__': main()