洛谷 P1255 数楼梯

题目链接：https://www.luogu.com.cn/problem/P1255

#include <iostream>
#include <string>
#include <vector>

using namespace std;

vector<int> add(vector<int> &a, vector<int> &b)
{
    // 保证a.size() >= b.size()
    if (a.size() < b.size())
    {
        return add(b, a);
    }
    
    vector<int> c;
    int t = 0;
    for (int i = 0; i < a.size(); i++)
    {
        // 逐位相加
        t += a[i];
        if (i < b.size())
        {
            t += b[i];
        }
        c.push_back(t % 10);
        t /= 10; // 向高位进位
    }
    
    // 如果最高位还有进位
    if (t)
    {
        c.push_back(t);
    }
    
    return c;
}

int main()
{
    int n;
    cin >> n;
    
    if (n <= 2)
    {
        cout << n;
        return 0;
    }
    
    vector<int> a, b, c;
    a.push_back(1);
    b.push_back(2);
    
    for (int i = 3; i <= n; i++)
    {
        c = add(a, b);
        a = b;
        b = c;
    }
    
    // 逆序输出
    for (int i = c.size() - 1; i >= 0; i--)
    {
        cout << c[i];
    }

    return 0;
}