字节跳动2020年九月笔试题-爬楼梯----------------原创

字节跳动2020年九月笔试题-爬楼梯

爬 n 层楼梯 每次只能爬一步或者两步，如果爬了一个两步则接下来不能爬两步，只能爬一步，求最多有多少种方法

笔试的时候感觉好复杂，后面自习思考之后，用状态转移（动态规划）+递归实现了

思路：f(n) = f(n-1)+f(n-2)

如果是f(n-2),说明爬了两步到达n层，则需要记录该状态，他的上一步只能是爬一步；

如果是f(n-1)，说明爬了一步步到达n层，记录该状态，他的上一步可以是一步或者两步；

综上：

f(n, status) = f(n-1, 1)+f(n-2,2);

利用递归求值；

话不多说，上代码：

#include <iostream>
using namespace std;

long CountNumber(int n, int status)
{
    if (n < 1)
    {
        return 0;
    }
    if (n == 1)
    {
        return 1;
    }
    if (n == 2)
    {
        //上一步是走的两步， 则只能全走一步
        if (status == 2)
        {
            return 1;
        }
        //上一步是走的1步， 则可以全走一步，或者直接走两步
        if (status == 1)
        {
            return 2;
        }
    }

    
    if (n > 2)
    {
        if (status == 0)
        {
            return CountNumber(n - 1, 1) + CountNumber(n - 2, 2);   
        }
        if (status == 1)
        {
            return CountNumber(n - 1, 1) + CountNumber(n - 2, 2);
        }
        if (status == 2)
        {
            return CountNumber(n - 1, 1);
        }
    }
}

int main(int argc, char const *argv[])
{
    int n;
    cin >> n;
    long ret;

    ret = CountNumber(n, 0);
    cout << ret << endl;
    return 0;
}