結果
| 問題 |
No.1785 Inequality Signs
|
| コンテスト | |
| ユーザー |
H20
|
| 提出日時 | 2021-12-14 12:41:44 |
| 言語 | PyPy3 (7.3.15) |
| 結果 |
AC
|
| 実行時間 | 400 ms / 2,000 ms |
| コード長 | 805 bytes |
| コンパイル時間 | 534 ms |
| コンパイル使用メモリ | 82,964 KB |
| 実行使用メモリ | 131,136 KB |
| 最終ジャッジ日時 | 2024-07-23 14:54:42 |
| 合計ジャッジ時間 | 13,991 ms |
|
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
| ファイルパターン | 結果 |
|---|---|
| sample | AC * 2 |
| other | AC * 52 |
ソースコード
N,K = map(int, input().split())
mod = 10**9+7
#コンビネーション
def cmb(n, r, mod):
if ( r<0 or r>n ):
return 0
r = min(r, n-r)
return g1[n] * g2[r] * g2[n-r] % mod
g1 = [1, 1] # 元テーブル
g2 = [1, 1] #逆元テーブル
inverse = [0, 1] #逆元テーブル計算用テーブル
for i in range( 2, N + 1 ):
g1.append( ( g1[-1] * i ) % mod )
inverse.append( ( -inverse[mod % i] * (mod//i) ) % mod )
g2.append( (g2[-1] * inverse[-1]) % mod )
#逆元、割り算する場合に使用
def modinv(a, mod=10**9+7):
return pow(a, mod-2, mod)
KF = [1,K]
for i in range(1,N):
KF.append(KF[-1]*(K-i)*modinv(i+1)%mod)
ans = 0
for i in range(1,N+1):
if i<=K:
m = N-i
ans = (ans + KF[i]*pow(2,i-1,mod)*cmb(m+i-1,m,mod))%mod
print(ans)
H20