HDOJ 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

 

这个题不难，重要的是找出规律，如果想不出来在纸上画画，画到8,9年规律就很清晰了，前四年，每年出生一头，从第五年开始，第n年出生的牛数等于 n-1年出生数+n-3年出生数，最后算总数就可以了。

#include<iostream>

using namespace std;

int main()
{
    int n;
    while(cin>>n&&n)
    {
        if(n<=4)
            cout<<n<<endl;
        else
        {
            int a[56],sum=0;
            a[0]=1; a[1]=1; a[2]=1; a[3]=1;
            for(int i=4;i<n;i++)
            {
                a[i]=a[i-1]+a[i-3];
            }
            for(int i=0;i<n;i++)
                sum+=a[i];
            cout<<sum<<endl;
        }
    }
}