結果

問題 No.533 Mysterious Stairs
ユーザー daleksprinter
提出日時 2018-09-11 16:24:05
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 14 ms / 5,000 ms
コード長 784 bytes
コンパイル時間 1,535 ms
コンパイル使用メモリ 167,692 KB
実行使用メモリ 26,980 KB
最終ジャッジ日時 2024-06-23 16:22:46
合計ジャッジ時間 2,398 ms
ジャッジサーバーID
(参考情報)
judge5 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 28
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
using namespace std;

#define rep(i,a,b) for(int i=a;i<b;i++)
#define int long long
typedef long long ll;
const ll inf = 100100100100100;
const int mod = 1000000007;


signed main(){
    std::ios::sync_with_stdio(false);
    std::cin.tie(0);

    int n;
    cin >> n;

    int dp1[n+1];
    int dp2[n+1];
    int dp3[n+1];

    if(n < 3){
        cout << 1 << endl;
        return 0;
    }

    rep(i,0,n+1){
        dp1[i] = 0; dp2[i] = 0; dp3[i] = 0;
    }

    dp1[1] = 1;
    dp2[2] = 1;
    dp1[3] = 1;
    dp2[3] = 1;
    dp3[3] = 1; 

    rep(i,4,n+1){
        dp1[i] = (dp2[i-1] + dp3[i-1]) % mod;
        dp2[i] = (dp1[i-2] + dp3[i-2]) % mod;
        dp3[i] = (dp1[i-3] + dp2[i-3]) % mod;
    }
    cout << (dp1[n] + dp2[n] + dp3[n])%mod << endl;


}
0