結果
問題 |
No.1127 変形パスカルの三角形
|
ユーザー |
![]() |
提出日時 | 2020-07-22 11:17:40 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 1,284 ms / 1,500 ms |
コード長 | 764 bytes |
コンパイル時間 | 725 ms |
コンパイル使用メモリ | 82,560 KB |
実行使用メモリ | 98,092 KB |
最終ジャッジ日時 | 2024-06-11 07:07:28 |
合計ジャッジ時間 | 18,490 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 30 |
ソースコード
def cmb(n, r): if n - r < r: r = n - r if r == 0: return 1 if r == 1: return n numerator = [n-r+k+1 for k in range(r)] denominator = [k+1 for k in range(r)] for p in range(2,r+1): pivot = denominator[p - 1] if pivot > 1: offset = (n - r) % p for k in range(p-1,r,p): numerator[k - offset] /= pivot denominator[k] /= pivot result = 1 for k in range(r): if numerator[k] > 1: result *= int(numerator[k]) return result mod=10**9+7 a,b=map(int,input().split()) n,k=map(int,input().split()) def answer(s,p): return (b*cmb(s-1,p-2)%mod+a*cmb(s-1,p-1)%mod)%mod print(answer(n,k)) print((a*answer(2*n-1,n)%mod+b*answer(2*n-1,n+1)%mod)%mod)