結果
問題 |
No.1011 Infinite Stairs
|
ユーザー |
👑 ![]() |
提出日時 | 2025-08-05 22:02:25 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 132 ms / 2,000 ms |
コード長 | 1,221 bytes |
コンパイル時間 | 198 ms |
コンパイル使用メモリ | 82,636 KB |
実行使用メモリ | 145,632 KB |
最終ジャッジ日時 | 2025-08-05 22:02:30 |
合計ジャッジ時間 | 4,540 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 24 |
ソースコード
""" https://yukicoder.me/problems/no/1011 包除の方が楽そう KをNこの箱に分ける - 1個以上は置く - max D個 """ #逆元 def inverse(a,mod): #aのmodを法にした逆元を返す return pow(a,mod-2,mod) #modのn!と、n!の逆元を格納したリストを返す(拾いもの) #factorialsには[1, 1!%mod , 2!%mod , 6!%mod… , n!%mod] が入っている #invsには↑の逆元が入っている def modfac(n, MOD): f = 1 factorials = [1] for m in range(1, n + 1): f *= m f %= MOD factorials.append(f) inv = pow(f, MOD - 2, MOD) invs = [1] * (n + 1) invs[n] = inv for m in range(n, 1, -1): inv *= m inv %= MOD invs[m - 1] = inv return factorials, invs def modnCr(n,r): #上で求めたfacとinvsを引数に入れるべし(上の関数で与えたnが計算できる最大のnになる) return fac[n] * inv[n-r] * inv[r] % mod mod = 10**9+7 fac,inv = modfac(10**6,mod) N,D,K = map(int,input().split()) ans = 0 K -= N for over in range(N+1): rem = K - D * over if rem < 0: continue ans += modnCr(rem+N-1,N-1) * pow(-1,over,mod) * modnCr(N,over) ans %= mod print (ans)