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

题目:

母牛从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)

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值