HDU 2018 母牛的故事 简单DP

母牛的故事 HDU - 2018

一头母牛，每年生一头母牛，小母牛长到第四年开始生小母牛，问第 $n$ 年有多少头母牛？

设 $dp[i]$ 为 $i$ 年的数量，则 $dp[1]=1,dp[2]=2,dp[3]=3$，$i>3$ 时有递推关系：

$$dp[i]=dp[i-1]+dp[i-3]$$

代码如下：

#include<iostream>
#include<cstdio>
//#define WINE
using namespace std;
int n,dp[56]={0,1,2,3};
int main(){
#ifdef WINE
    freopen("data.in","r",stdin);
#endif
    for(int i=4;i<=55;i++)
        dp[i]=dp[i-1]+dp[i-3];
    while(scanf("%d",&n)==1){
        if(n==0)break;
        printf("%d\n",dp[n]);
    }
    return 0;
}