问题 1004: [递归]母牛的故事

原题目链接 --蓝桥杯练习题目
问题 1004: [递归]母牛的故事

时间限制: 1Sec 内存限制: 128MB 提交: 41370 解决: 12438

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

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

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

样例输入
2
4
5
0
样例输出
2
4
6

#include <cstdio>
#include <iostream>
using namespace std;
// int func(int n) 超时
// {
//   if(n <= 4) return n;
//   else return func(n-1) + func(n-3);
// }
int main()
{
  int n = -1;
  int ans;
  int a[55];
  a[1] = 1;a[2] = 2;a[3] = 3;a[4] = 4;
  for(int i=5;i<=54;i++)
    a[i] = a[i-1] + a[i-3];
  cin>>n;
  while(n != 0)
  {
    if(n<=4) ans = a[n];
    else ans = a[n-1] + a[n-3];
    cout<<ans<<endl;
    cin>>n;
  }
  return 0;
}