递推+高精度+滚动数组

数楼梯

题目描述

楼梯有 $N$ 阶，上楼可以一步上一阶，也可以一步上二阶。

编一个程序，计算共有多少种不同的走法。

输入格式

一个数字，楼梯数。

输出格式

输出走的方式总数。

样例 #1

样例输入 #1

4

样例输出 #1

5

提示

• 对于 $60\%$ 的数据，$N\le 50$；
• 对于 $100\%$ 的数据，$1\le N\le 5000$。

题解：

#include <iostream>
#include <string>
using namespace std;
int N; int f[2][5003], len = 0, pre1 = 0, pre2 = 1;;

void add(int k) {
    for (int i = 1; i <= len; i++) {
        f[pre1][i] = f[pre1][i] + f[pre2][i];
    }
    for (int i = 1; i <= len; i++) {
        if (f[pre1][i] >= 10) {
            f[pre1][i+1] += f[pre1][i] / 10;
            f[pre1][i] %= 10;
            if (i + 1 > len)
                len++;
        }
    }
}

int main()
{
    cin >> N;
    len = 1;
    f[0][1] = 1, f[1][1] = 2;
    if (N < 3) {
        cout << f[N - 1][1];
        return 0;
    }
    for (int i = 3; i <= N; i++) {
        add(i); 
        pre1 = (pre1 + 1) % 2;
        pre2 = (pre2 + 1) % 2;
    }
    for (int i = len; i > 0; i--)
        cout << f[pre2][i];
    return 0;
}

这题乍一看，递推关系非常好找，假设f[k]为走到第k个台阶的方法数，那么就有：

f[k] = f[k-1] + f[k-2] (k>=3)

然后根据这个递推式，用递归就可以写出来。我一开始就是这么想的，但是这样做却TLE了，看了一下题解才知道，这题的数据只能用高精度。。。

所以，撸了一下高精度，由于递推只需要用到第k-1和第k-2个数据，所以可以考虑使用两个数组来滚动，用pre1和pre2来标记k-2和k-1的数据在哪一行。题解里直接暴力开O(n2)的空间，感觉有点不适，所以优化了这一下下。。。