結果

問題 No.1127 変形パスカルの三角形
ユーザー 37zigen
提出日時 2020-07-28 16:34:17
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 148 ms / 1,500 ms
コード長 541 bytes
コンパイル時間 123 ms
コンパイル使用メモリ 82,380 KB
実行使用メモリ 101,548 KB
最終ジャッジ日時 2024-06-28 20:46:31
合計ジャッジ時間 5,136 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff # 

MOD=10**9+7
MAX=10**6
fac=[0]*MAX
ifac=[0]*MAX
inv=[0]*MAX

fac[0]=fac[1]=ifac[0]=ifac[1]=inv[0]=inv[1]=1
for i in range(2,MAX):
    fac[i]=i*fac[i-1]%MOD
    inv[i]=MOD-(MOD//i)*inv[MOD%i]%MOD
    ifac[i]=inv[i]*ifac[i-1]%MOD

def C(n,k):
    if n<0 or k<0 or n-k<0:
        return 0
    return fac[n]*ifac[n-k]*ifac[k]%MOD

a,b=map(int,input().split())
n,k=map(int,input().split())

# 0<=k<=n

def f(n,k):
    return (C(n,k)*a+C(n,n+1-k)*b)%MOD
ans=0
for i in range(1,n+2):
    ans+=f(n-1,i-1)**2
    ans%=MOD
print(f(n-1,k-1))
print(ans)
0