結果

問題 No.533 Mysterious Stairs
ユーザー @abcde@abcde
提出日時 2019-05-23 23:46:49
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 13 ms / 5,000 ms
コード長 1,125 bytes
コンパイル時間 1,512 ms
コンパイル使用メモリ 166,512 KB
実行使用メモリ 26,736 KB
最終ジャッジ日時 2024-09-17 09:55:19
合計ジャッジ時間 2,178 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 28
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
using LL = long long;
const LL MOD = 1e9 + 7;
const int MAX = 1e6;
LL dp[MAX + 1][3];

int main() {
    
    // 1. 入力情報取得.
    LL N;
    cin >> N;
    
    // 2. jumpの仕方をカウント.
    dp[0][0] = 1, dp[1][1] = 1, dp[2][0] = 1, dp[2][1] = 1, dp[2][2] = 1;
    for(int i = 3; i < N; i++){
        // (i + 1)段目 に 到達するパターンを記載.

        // 2-1. i段目 に, 2 step or 3 step で到達していた場合.
        dp[i][0] = dp[i - 1][1] + dp[i - 1][2], dp[i][0] %= MOD;

        // 2-2. (i - 1)段目 に, 1 step or 3 step で到達していた場合.
        dp[i][1] = dp[i - 2][0] + dp[i - 2][2], dp[i][1] %= MOD;

        // 2-3. (i - 2)段目 に, 1 step or 2 step で到達していた場合.
        dp[i][2] = dp[i - 3][0] + dp[i - 3][1], dp[i][2] %= MOD;
    }

    // 3. 出力 ～ 後処理.
    // ex.
    // N = 1000000 ならば, 480942662 で OK?
    LL ans = 0;
    ans += dp[N - 1][0], ans %= MOD;
    ans += dp[N - 1][1], ans %= MOD;
    ans += dp[N - 1][2], ans %= MOD;
    cout << ans << endl;
    return 0;
    
}
0