結果

問題 No.1011 Infinite Stairs
ユーザー H20
提出日時 2021-06-25 16:25:50
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 851 ms / 2,000 ms
コード長 365 bytes
コンパイル時間 195 ms
コンパイル使用メモリ 82,160 KB
実行使用メモリ 209,680 KB
最終ジャッジ日時 2024-07-17 12:18:47
合計ジャッジ時間 8,272 ms
ジャッジサーバーID
(参考情報)
judge4 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 24
権限があれば一括ダウンロードができます

ソースコード

diff #

import itertools
N,d,K = map(int,input().split())
mod = 10**9+7
DP = [0]*((N+2)*d)
DP[0] = 1
lenDP = len(DP)
for _ in range(N):
    TEMP = [0]*(lenDP)
    for i in range(lenDP):
        if DP[i]>=1:
            TEMP[i+1]+=DP[i]
            TEMP[i+d+1]-=DP[i]
    DP = list(itertools.accumulate(TEMP))
    for i in range(lenDP):
        DP[i]=DP[i]%mod
print(DP[K])
0