一个好玩的编程小游戏—— 母牛生小牛

题目：

母牛从3~7岁初每年会生产1头小母牛，10岁后死亡（10岁任然存活），假设初始有一头刚出生的母牛，请问第n年有多少头母牛？（年从第一年开始计数）

注：第三年初会出生 第一头母牛，故第三年有两头母牛。

第五年初，第三年出生的母牛会生产，故第五年有五头母牛。

岁数是虚数

示例：
输入 输出
2 1
3 2
4 3
5 5
12 123

分析：

为计算第n年的牛数，我们需要知道每头牛的年龄。年龄不需要存储和更新，只记录每年有多少头小牛出生，即可推算得到。

创建列表，记录每年牛的总数和当年出生的小牛数[all_nums, birth_num]；
第n年新生小牛数 = 处在生育年龄段内小牛牛数总和
第n年的牛数 = 10年内出生的小牛总和

代码：

        import numpy as np
        # 输入要计算牛数的年份
        n = int(input())
        #n = 12
        #创建列表，记录每年牛的总数和当年出生的小牛数[all_nums, birth_num]；
        nums = []
        # 初始化第1年小牛数：共1头牛，新出生1头牛
        nums.append([1, 1])
        nums = np.array(nums)
         
        # 计算往后每年的牛数
        for year in range(1, n):
            #第n年新生小牛数 = 处在生育年龄段内小牛牛数总和
            if year >= 7:
                birth_num = sum(nums[-6:-1, 1])   #[:][-6:-1])
            else:
                birth_num = sum(nums[:-1, 1])
                
            #第n年的牛数 = 10年内出生的小牛总和
            if year >= 10:
                all_nums = sum(nums[-9:, 1]) + birth_num
            else:
                all_nums = sum(nums[:, 1]) + birth_num
            new = np.array([all_nums, birth_num])
            nums = np.row_stack((nums, new))
        print(nums[-1, 0])
        #print(nums)