2019【算法】【字节跳动】【笔试】

一. 母牛生小牛

题目：

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