結果

問題 No.1011 Infinite Stairs
ユーザー 沙耶花
提出日時 2020-03-20 21:38:35
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 312 ms / 2,000 ms
コード長 678 bytes
コンパイル時間 3,089 ms
コンパイル使用メモリ 197,940 KB
最終ジャッジ日時 2025-01-09 08:23:40
ジャッジサーバーID
(参考情報)
judge5 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 24
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
using namespace std;
#define modulo 1000000007
#define mod(mod_x) ((((long long)mod_x+modulo))%modulo)
#define Inf 1000000000


int main(){
    
	int N,d,K;
	cin>>N>>d>>K;
	
	vector<vector<int>> dp(2,vector<int>(K+1,0));
	dp[0][0] = 1;
	
	for(int i=0;i<N;i++){
		for(int j=1;j<=K;j++){
			dp[0][j] = mod(dp[0][j] + dp[0][j-1]);
		}
		
		for(int j=1;j<=K;j++){
			dp[1][j] = dp[0][j-1];
			if(j-d-1>=0){
				dp[1][j] = mod(dp[1][j] - dp[0][j-d-1]);
			}
		}
		/*
		for(int j=0;j<=K+1;j++){
			cout<<dp[1][j]<<',';
		}
		cout<<endl;
		*/
		swap(dp[0],dp[1]);
		for(int j=0;j<=K;j++){
			dp[1][j] = 0;
		}
	}
	
	cout<<dp[0].back()<<endl;
	
    return 0;
}
0