結果

問題 No.1011 Infinite Stairs
ユーザー ktr216
提出日時 2020-03-20 22:28:17
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 502 ms / 2,000 ms
コード長 1,129 bytes
コンパイル時間 1,603 ms
コンパイル使用メモリ 171,792 KB
実行使用メモリ 427,536 KB
最終ジャッジ日時 2024-07-17 11:52:03
合計ジャッジ時間 4,490 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 24
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include "bits/stdc++.h"
#define rep(i,n) for(int i=0;i<n;i++)
using namespace std;

typedef long long ll;

int main() {
    ll N, d, K, MOD =1e9 + 7;
    cin >> N >> d >> K;
    //dp[i][j] : i回の移動でj段目にたどり着く方法
    //ds[i][j] : i回の移動でj段目以下にたどり着く方法
    vector<vector<ll>> dp(N + 1, vector<ll>(K + 1, 0)), ds(N + 1, vector<ll>(K + 1, 0));
    dp[0][0] = 1;
    rep(j, K + 1) ds[0][j] = 1;
    rep(i, N) {
        rep(j, K) {
            if (j >= d) {
                dp[i + 1][j + 1] = ds[i][j] - ds[i][j - d] + MOD;
                dp[i + 1][j + 1] %= MOD;
            } else {
                dp[i + 1][j + 1] = ds[i][j];
            }
        }
        rep(j, K) {
            ds[i + 1][j + 1] = ds[i + 1][j] + dp[i + 1][j + 1];
            ds[i + 1][j + 1] %= MOD;
        }
    }
    /*
    rep(i, N + 1) {
        rep(j, K + 1) {
            cout << dp[i][j] << " ";
        }
        cout << "\n";
    }
    rep(i, N + 1) {
        rep(j, K + 1) {
            cout << ds[i][j] << " ";
        }
        cout << "\n";
    }
    */
    cout << dp[N][K] << "\n";
}
0