結果
問題 | No.1127 変形パスカルの三角形 |
ユーザー |
![]() |
提出日時 | 2020-07-23 16:31:51 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 1,186 ms / 1,500 ms |
コード長 | 745 bytes |
コンパイル時間 | 557 ms |
コンパイル使用メモリ | 82,172 KB |
実行使用メモリ | 99,056 KB |
最終ジャッジ日時 | 2024-06-23 17:47:16 |
合計ジャッジ時間 | 16,364 ms |
ジャッジサーバーID (参考情報) |
judge5 / 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 ans(x,y): return (a*cmb(x-1,y-1)+b*cmb(x-1,y-2))%mod print(ans(n,k)) print((a*ans(2*n-1,n)+b*ans(2*n-1,n+1))%mod)