結果

問題 No.1127 変形パスカルの三角形
ユーザー 37zigen
提出日時 2020-07-28 16:31:04
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
TLE  
実行時間 -
コード長 541 bytes
コンパイル時間 83 ms
コンパイル使用メモリ 12,416 KB
実行使用メモリ 128,500 KB
最終ジャッジ日時 2024-06-28 20:45:52
合計ジャッジ時間 50,728 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1 TLE * 1
other AC * 10 TLE * 20
権限があれば一括ダウンロードができます

ソースコード

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