結果
問題 |
No.1011 Infinite Stairs
|
ユーザー |
![]() |
提出日時 | 2020-05-18 23:43:01 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 108 ms / 2,000 ms |
コード長 | 693 bytes |
コンパイル時間 | 270 ms |
コンパイル使用メモリ | 12,800 KB |
実行使用メモリ | 17,920 KB |
最終ジャッジ日時 | 2024-07-17 12:07:32 |
合計ジャッジ時間 | 2,337 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 24 |
ソースコード
N, d, K = map(int, input().split()) mod = int(1e9)+7 def inved(a): x, y, u, v, k, l = 1, 0, 0, 1, a, mod while l != 0: x, y, u, v = u, v, x - u * (k // l), y - v * (k // l) k, l = l, k % l return x % mod fact = [1 for _ in range(N+K+1)] invf = [1 for _ in range(N+K+1)] for i in range(N+K): fact[i+1] = (fact[i]*(i+1))%mod invf[-1] = inved(fact[-1]) for i in range(N+K, 0, -1): invf[i-1] = (invf[i] * i) % mod S = 0 sign = 1 for i in range(N+1): if K-i*d-N < 0: break else: S += (sign * fact[K-1-i*d] * invf[i] * invf[N-i] * invf[K-i*d-N]) % mod S %= mod sign *= mod - 1 sign %= mod print((N*S)%mod)