杭电oj HDOJ 2018 母牛的故事

杭电oj HDOJ 2018 母牛的故事

Problem Description

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

Input

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

Output

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

解题思路

前3年可以很容易地算出：第一年为1头，第二年为2头，第三年为3头。
从第四年开始，母牛的个数是前一年母牛的个数与三年前母牛个数之和。
因为第一头母牛所生出的小牛，在其出生的第四个年头会生出一个小小牛，并且在此期间，最开始的那头母牛，还会每年生出一头小牛

本人的C++解决方案

#include <iostream>
using namespace std;

int main()
{
    int n, i, num[54];
    num[0] = 1;
    num[1] = 2;
    num[2] = 3;
    for (i = 3; i < 54; i++) {
        num[i] = num[i - 1] + num[i - 3];
    }
    while (cin>>n) {
        if (n == 0) {
            break;
        }
        if (n <= 0 || n >= 55) {
            continue;
        }
        cout<<num[n - 1]<<endl;
    }
    return 0;
}

代码通过HDOJ平台运行检查，如发现错误，欢迎指出和纠正，谢谢！