結果

問題 No.1011 Infinite Stairs
ユーザー nagitaosunagitaosu
提出日時 2020-03-20 22:31:29
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 579 ms / 2,000 ms
コード長 548 bytes
コンパイル時間 199 ms
コンパイル使用メモリ 82,012 KB
実行使用メモリ 270,548 KB
最終ジャッジ日時 2024-07-17 11:52:30
合計ジャッジ時間 4,531 ms
ジャッジサーバーID
(参考情報)
judge4 / judge5
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 24
権限があれば一括ダウンロードができます

ソースコード

diff #

#!/usr/bin/env python3
import sys
input = sys.stdin.readline
MOD = 10**9 + 7

n, d, k = map(int, input().split())
if n == k:
    print(1)
    exit()
k -= n
d -= 1
dp = [[0] * (k + 1) for _ in range(n)]
for i in range(min(k + 1, d + 1)):
    dp[0][i] = 1
for i in range(n-1):
    for j in range(k+1):
        if j - 1 - d >= 0:
            dp[i + 1][j] = dp[i + 1][j-1] + dp[i][j] - dp[i][j - 1 - d]
            dp[i + 1][j] %= MOD
        else:
            dp[i + 1][j] = dp[i + 1][j-1] + dp[i][j]
            dp[i + 1][j] %= MOD

print(dp[-1][-1])
0