結果

問題 No.1127 変形パスカルの三角形
ユーザー flippergo
提出日時 2024-12-13 18:41:19
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 339 ms / 1,500 ms
コード長 514 bytes
コンパイル時間 320 ms
コンパイル使用メモリ 82,356 KB
実行使用メモリ 80,364 KB
最終ジャッジ日時 2024-12-13 18:41:29
合計ジャッジ時間 8,437 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff # 

a,b = map(int,input().split())
n,k = map(int,input().split())
MOD = 10**9+7
A = [1]*(n+1)
for i in range(1,n+1):
    A[i] = (A[i-1]*i)%MOD
B = [1]*(n+1)
B[n] = pow(A[n],MOD-2,MOD)
for i in range(n-1,0,-1):
    B[i] = (B[i+1]*(i+1))%MOD
def comb(s,t):
    if t<0 or t>s:
        return 0
    if t==0 or t==s:
        return 1
    return (A[s]*B[t]*B[s-t])%MOD
ans = a*comb(n-1,k-1)+b*comb(n-1,n+1-k)
print(ans%MOD)
cnt = 0
for j in range(1,n+2):
    cnt = (cnt+(comb(n-1,j-1)*a+comb(n-1,n+1-j)*b)**2)%MOD
print(cnt)
0