結果
問題 |
No.1127 変形パスカルの三角形
|
ユーザー |
![]() |
提出日時 | 2023-10-23 02:32:39 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 114 ms / 1,500 ms |
コード長 | 543 bytes |
コンパイル時間 | 314 ms |
コンパイル使用メモリ | 82,384 KB |
実行使用メモリ | 118,500 KB |
最終ジャッジ日時 | 2024-09-22 09:54:30 |
合計ジャッジ時間 | 5,073 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 30 |
ソースコード
a,b=map(int,input().split()) n,k=map(int,input().split()) mod=10**9+7 FACT=[1] for i in range(1,3*10**5+1): FACT.append(FACT[-1]*i%mod) FACT_INV=[pow(FACT[-1],mod-2,mod)] for i in range(3*10**5,0,-1): FACT_INV.append(FACT_INV[-1]*i%mod) FACT_INV.reverse() def Combi(a,b): if 0<=b<=a: return FACT[a]*FACT_INV[b]%mod*FACT_INV[a-b]%mod else: return 0 print((Combi(n-1,k-1)*a+Combi(n-1,k-2)*b)%mod) ANS=0 for k in range(1,n+2): ANS+=((Combi(n-1,k-1)*a+Combi(n-1,k-2)*b)%mod)**2 ANS%=mod print(ANS)