結果
問題 | 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)