题解 | 兔子的数量

def rabbit_count(n):
    # 将输入转换为整数
    n = int(n)
    # 初始化前两个兔子数量
    if n == 1:
        return 2
    elif n == 2:
        return 3
    # 初始化兔子数量列表
    ls1 = [2, 3]
    # 从第 3 个月开始递推
    for i in range(2, n):
        # 计算当前月的兔子数量
        num3 = ls1[i - 1] + ls1[i - 2]
        # 将当前月的兔子数量添加到列表中
        ls1.append(num3)
    # 返回第 n 个月的兔子数量
    return ls1[-1]

# 获取用户输入并调用函数
print(rabbit_count(input()))