C++描述杭电OJ 2018.母牛的故事 ||

C++描述杭电OJ 2018.母牛的故事 ||

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

Input

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

Output

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

Sample Input
2
4
5
0

Sample Output
2
4
6

代码实现

#include<iostream>
using namespace std;

class Solution
{
public :
    void solution()
    {
        //找出规律，sum[]用来存每一年总共牛数量,sum[0]作为第0年
        int n,sum[10000]= {0,1,2,3,4};
        while(cin >> n && n)
        {
            //从第5年开始，每一年的大前年的新生牛开始产生新牛崽
            if(n>=5)
            {
                for(int i=5; i<=n; i++)
                {
                    //每一年的总牛数=前一年的总牛数+大前年的总牛数,sum[i]表示第i年总牛数
                    sum[i]=sum[i-1]+sum[i-3];
                }
            }
            cout << sum[n] << endl;
        }
    }
};
int main()
{
    Solution solu;
    solu.solution();
    return 0;
}