hdu 2018 母牛的故事(DP)

Problem Description

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

Input

输入数据由多个测试实例组成，每个测试实例占一行，包括一个整数n(0<n<55)，n的含义如题目中描述。
n=0表示输入数据的结束，不做处理。

Output

对于每个测试实例，输出在第n年的时候母牛的数量。
每个输出占一行。

Sample Input

2 4 5 0

Sample Output

2 4 6

 

解题思路：

       dp[i] 代表第 i 年的母牛数量，dp[i-3] 代表第 i 年可以生下小母牛的母牛数量，因此状态方程为dp[i] = dp[i-1] + dp[i-3];

#include<bits/stdc++.h>
using namespace std;

int main(){
    int n;
    int dp[100];
    dp[1] = 1;
    for(int i = 2; i < 55; i++){
        if(i < 4){
            dp[i] = dp[i-1] + 1;
        }else{
            dp[i] = dp[i-1] + dp[i-3];
        }
    }

    while(cin >> n && n != 0){
        cout << dp[n] << endl;
    }
    return 0;
}