トップページに戻る    次の競技プログラミングの問題へ    前の競技プログラミングの問題へ

No.533 Mysterious Stairs

■■■問題■■■

赳也君は本日、ある雑居ビルの5Fで面接があります。
運動不足解消の為、階段を使って面接会場まで向かいます。
階段数はちょうどNstepあります。
1回のjumpで1 or 2 or 3 step登れます。

雑居ビル全体に奇妙なオーラが流れていて(もしくは面接への緊張感等の心理的理由から)、
同じstep数のjumpを連続して行う事は出来ません。

赳也君の面接会場までのjumpの仕方は何通りあるでしょうか。
答えが非常に大きくなる場合があるので、1000000007で割った余りを求めなさい。

■■■入力■■■

N

1行目に面接会場までのstep数Nが与えられます。1 <= N <= 100万

■■■出力■■■

ちょうどNstepするjumpのしかたが何通りあるかを出力してください。
最後に改行してください。


C#のソース

using System;
using System.Collections.Generic;

class Program
{
    static string InputPattern = "Input1";

    static List<string> GetInputList()
    {
        var WillReturn = new List<string>();

        if (InputPattern == "Input1") {
            WillReturn.Add("2");
            //1
            //1回のjumpで2stepする方法しかありません。
            //1step+1stepという連続した同step数のjumpは行う事が出来ません。
        }
        else if (InputPattern == "Input2") {
            WillReturn.Add("4");
            //3
            //(1,2,1),(3,1),(1,3)の3通りです
        }
        else if (InputPattern == "Input3") {
            WillReturn.Add("49993");
            //407466683
        }
        else {
            string wkStr;
            while ((wkStr = Console.ReadLine()) != null) WillReturn.Add(wkStr);
        }
        return WillReturn;
    }

    static void Main()
    {
        const int Hou = 1000000007;

        List<string> InputList = GetInputList();
        int N = int.Parse(InputList[0]);

        //場合の数[現在位置,前のStep数]なDP表
        int[,] DPArr = new int[N + 1, 4];
        DPArr[0, 0] = 1;

        for (int I = 0; I <= N; I++) {
            for (int J = 0; J <= 3; J++) {
                if (DPArr[I, J] == 0) continue;

                //歩数の候補
                var HosuuList = new List<int>() { 1, 2, 3 };
                HosuuList.Remove(J);

                foreach (int EachHosuu in HosuuList) {
                    int NewI = I + EachHosuu;
                    if (NewI > N) break;

                    int NewJ = EachHosuu;

                    DPArr[NewI, NewJ] += DPArr[I, J];
                    DPArr[NewI, NewJ] %= Hou;
                }
            }
        }

        int Answer = 0;
        for (int J = 1; J <= 3; J++) {
            Answer += DPArr[N, J];
            Answer %= Hou;
        }
        Console.WriteLine(Answer);
    }
}


解説

動的計画法で解いてます。