結果

問題 No.1011 Infinite Stairs
ユーザー snrnsidy
提出日時 2021-10-28 13:33:08
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 561 ms / 2,000 ms
コード長 926 bytes
コンパイル時間 2,333 ms
コンパイル使用メモリ 191,788 KB
最終ジャッジ日時 2025-01-25 08:01:51
ジャッジサーバーID
(参考情報)
judge2 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 24
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>

using namespace std;

const long long int MOD = 1e9 + 7;
long long int dp[301][90001];
long long int psum[90001];

int main(void)
{
    cin.tie(0);
    ios::sync_with_stdio(false);

    int n,d,k;

    cin >> n >> d >> k;

    memset(dp,0,sizeof(dp));

    dp[0][0] = 1;
    for(int i=0;i<n;i++)
    {
        memset(psum,0,sizeof(psum));
        psum[0] = dp[i][0];
        for(int j=1;j<=k;j++)
        {
            psum[j] += psum[j-1];
            psum[j]%=MOD;
            psum[j] += dp[i][j];
            psum[j]%=MOD;
        }

        for(int j=1;j<=k;j++)
        {
            long long int val = psum[j-1];
            if(j-1-d >= 0)
            {
                val -= psum[j-1-d];
                val%=MOD;
                if(val < 0) val += MOD;
            }
            dp[i+1][j] += val;
            dp[i+1][j]%=MOD;
        } 
    }

    cout << dp[n][k] << '\n';
    return 0;
}
0