有一农场培育的母羊寿命是5年，母羊会在第2年和第4年各生下一头母羊，第5年会死去。假如农场一开始只有1头母羊（1岁），请用Python编写程序，可以求得n年后，该农场共有几只母羊？

def count_sheep(n):
    # 初始只有1头1岁的母羊
    sheep = {'1岁': 1, '2岁': 0, '3岁': 0, '4岁': 0, '5岁': 0}
    for i in range(n):
        # 第5年死亡
        sheep['5岁'] = sheep['4岁']
        sheep['4岁'] = sheep['3岁']
        sheep['3岁'] = sheep['2岁']
        # 第2年和第4年各生一头
        sheep['2岁'] = sheep['1岁']
        sheep['4岁'] += sheep['2岁']
        sheep['1岁'] = sheep['4岁']
    # 返回所有母羊数量之和
    return sum(sheep.values())

# 测试
print(count_sheep(10))  # 输出：842