题目
一个猪圈里有两只刚满2岁的猪，每只猪只能活四年，每只猪第四年还可以生一次崽，2岁后每两只猪每年可以生4头小猪（假设小猪性别比例总是1：1），那么n年后，猪圈里有几只猪？
解题思路
image.png
代码
aList = [2,2]
year = 0
n = int(input('输入第几年： '))
while year < n:
bList = aList
for i in range(len(aList)):
bList[i] = aList[i]+1 #增加1岁
geshu = bList.count(3)+bList.count(4) #统计3，4年的猪
bList.extend([1]*geshu*2) #根据3，4年猪的数量，添加1岁的猪
cList = [i for i in bList if i <= 4 ] #剔除超过4岁的猪
aList = cList
year += 1
print('猪圈里猪的个数：{}头\n猪的年龄：{}'.format(len(cList),cList))
验证
image.png
网友评论