【华为机试真题 Python】统计每个月兔子的总数

目录

题目描述

输入

输出

示例

参考代码

---

题目描述

有一种兔子，从出生后第3个月起每个月都生一只兔子，小兔子长到第三个月后每个月又生一只兔子。

例子：假设一只兔子第3个月出生，那么它第5个月开始会每个月生一只兔子。
一月的时候有一只兔子，假如兔子都不死，问第n个月的兔子总数为多少？

数据范围：
输入满足 1≤n≤31

输入

输入描述：
输入一个int型整数表示第n个月

输出

输出描述：
输出对应的兔子总数

示例

示例 1
输入：
3

输出：
2

参考代码

def dfs(m, h, maxh):
    if h > maxh:
        return 1
    elif h < 3:
        return dfs(1, h+1, maxh)
    else:
        return dfs(1, h+1, maxh) + dfs(1, 1, maxh-h+1)


def calc():
    month = int(input())
    print(dfs(1, 1, month))


while True: