绿巨人吃饼

题目：绿巨人要吃m个饼，他一口可以吃1个或者2个饼，但是每次一口2个之后的连续两口只能一口1个。请问绿巨人有几种方式吃完m个饼？

 

分析：

递推前几项或画二叉树可知 m <= 5 时，f(m) = m；

双饼奇下: 吃了2个饼之后连续两口只能一口1个。反观历史：状态dp[i-1]（吃1个）=dp[i-2]（吃1个）=dp[i-3]（吃2个）等价于dp[i-4]吃4个。

def f(m):
    """绿巨人吃完m个饼有多少种方式的函数f(m)"""
    if m < 5: # 画二叉树或递推前几项可知 m <= 5 时，f(m) = m
        return m
    # 动态规划
    dp = [0 for _ in range(m + 1)] # 初始化dp数组
    for i in range(5): # 当 m < 5 时，dp[i] = i
        dp[i] = i
    for i in range(5, m + 1): # 当 m >= 5 时，状态转移方程为：dp[i] = dp[i - 1] + dp[i - 4]
        dp[i] = dp[i - 1] + dp[i - 4]
    return dp[-1] # 返回dp数组最后一个元素


m = int(input()) # 输入非负整数m
print(f(m)) # 打印函数返回结果

# 动态规划
m = int(input()) # 输入非负整数m
dp = [i for i in range(0, m+1)] # 初始化dp数组(当 m <= 5 时，dp[i] = i)
for i in range(5, m+1): # 当 m >= 5 时，状态转移方程为：dp[i] = dp[i - 1] + dp[i - 4]
  dp[i] = dp[i - 1] + dp[i - 4]
print(dp[-1]) # 打印dp数组最后一个元素