题目

一个猪圈里有两只刚满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