結果
問題 | 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+7FACT=[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]%modelse:return 0print((Combi(n-1,k-1)*a+Combi(n-1,k-2)*b)%mod)ANS=0for k in range(1,n+2):ANS+=((Combi(n-1,k-1)*a+Combi(n-1,k-2)*b)%mod)**2ANS%=modprint(ANS)