【Python】母牛问题

Python 3.6.7 

母牛问题

题目描述：有一头母牛，它每年年初生一头小母牛。
每头小母牛从第 4 个年头开始，每年年初也生一头小母牛。（别问我公牛上哪去了……）
请编程实现计算在第 n 年的时候，共有多少头母牛？

输入：年份 n
输出：母牛的数量

示例：
第 1 年有 1 头母牛
第 2 年有 2 头母牛
第 3 年有 3 头母牛
第 4 年有 4 头母牛
第 5 年有 6 头母牛
第 6 年有 9 头母牛
第 7 年有 13 头母牛
第 8 年有 19 头母牛
第 9 年有 28 头母牛

def cow1(n):
    """循环计算
    big表示成熟母牛，small表示小母牛
    小母牛经过三年（从列表里走过3个位置）就变成成熟母牛
    成熟母牛有多少只当年就会新生多少小母牛（放在small最后）"""
    big, small = 0, [0, 1, 0]
    for _ in range(n):
        big += small.pop(0)
        small.append(big)
    return big + sum(small)


def cow21(n):
    """递归
    牛群总数 = 去年总数 + 今年新生儿数量
    今年新生儿数量 = 今