Python DP问题 母牛的故事

问题描述:

有一头母牛,它每年年初生一头小母牛。每头小母牛从第四个年头开始,每年年初也生一头小母牛。请编程实现在第n年的时候,共有多少头母牛?


# -*- coding: utf-8 -*-

# 当前的牛数 = 上一年的牛数 + 新生的牛数
# 新生的牛数 = 三年前的牛数


year_count = 5
DP_count = [1 for i in range(year_count)]


def cow_born():
    current_year = 1

    while current_year <= year_count:

        if current_year <= 4:
            DP_count[current_year-1] = current_year

        else:
            DP_count[current_year-1] = DP_count[current_year - 2] + DP_count[current_year - 4]

        current_year += 1

    return DP_count[-1]


if __name__ == '__main__':
    print cow_born()



©️2020 CSDN 皮肤主题: 深蓝海洋 设计师:CSDN官方博客 返回首页