一. 母牛生小牛
题目:
母牛从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年新生小牛数 = 处在生育年龄段内小牛牛数总和