走楼梯2

最新推荐文章于 2024-10-01 23:10:03 发布

qwer0408

最新推荐文章于 2024-10-01 23:10:03 发布

阅读量59

点赞数

文章标签：数据结构 leetcode 算法

本文链接：https://blog.csdn.net/qwer0408/article/details/129674530

版权

本体相对普通的走楼梯进行了一些变形，即一次性最多连走2次2步，那么我们可以引入一个二维数组来模拟走到一个楼梯的的状态（走了一次两步或两次两步）代码如下

for (int i = 1; i <= n; i++) {
            f[i][0] = f[i - 1][0] + f[i - 1][1] + f[i - 1][2]//下标为0表示走一步台阶上来的;
            if (i < 2) continue;
            f[i][1] = f[i - 2][0]//下标为1表示第一次走2步;
            f[i][2] = f[i - 2][1]//下标为2表示第二次走2步;
        }

之后就比较容易了，下面附上完整代码

#include <iostream>
using namespace std;
long long f[55][3];
int main()
{

        int n;
        cin >> n;
        f[0][0] = 1;
        for (int i = 1; i <= n; i++) {
            f[i][0] = f[i - 1][0] + f[i - 1][1] + f[i - 1][2];
            if (i < 2) continue;
            f[i][1] = f[i - 2][0];
            f[i][2] = f[i - 2][1];
        }
        cout << f[n][0] + f[n][1] + f[n][2];
        return 0;
}