题目:
母牛从3~7岁初每年会生产1头小母牛,10岁后死亡(10岁仍然存活),假设初始有一头刚出生的母牛,请问第n年有多少头母牛?(年从第一年开始计数)
注:第三年初会出生 第一头母牛,故第三年有两头母牛。
第五年初,第三年出生的母牛会生产,故第五年有五头母牛。(岁数是虚岁)
示例:
分析 :从第三年开始生小牛,十一岁死亡。我们可以利用一个列表记录下每一年1-11岁的牛的分别的个数。
第一年:[1, 0, 0, 0, 0, 0, 0, 0, 0, 0]
第二年:[0, 1, 0, 0, 0, 0, 0, 0, 0, 0]
第三年:[1, 0, 1, 0, 0, 0, 0, 0, 0, 0]
第四年:[1, 1, 0, 1, 0, 0, 0, 0, 0, 0]
第五年:[2, 1, 1, 0, 1, 0, 0, 0, 0, 0]
下一年等于上一年【1:】右移一位(因为小牛都长大了一岁),首位的小牛等于【2-7】小牛的和(因为小牛在2-7岁才可以生小牛)
代码实现如下:
def cow(n):
n_cows = [0 for i in range(1,11)]
n_cows[0] = 1
#print(n_cows)
for year in range(n-1):
n_cows[1:] = n_cows[:-1]
n_cows[0] = sum(n_cows[2:7])
#print(n_cows)
return sum(n_cows)