蓝桥杯 提高题 母牛的故事

蓝桥杯 提高题 母牛的故事

题目描述

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

输入

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

输出

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

样例输入

2
4
5
0

样例输出

2
4
6

这题我直接模拟的，没有推递推公式，如果比赛时想不出来可以直接这样模拟哦

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

int main()
{
    ll n,add[100];
    while(cin>>n && n){
        ll cow=1,ans=0;
        for(int i=1;i<=n;i++){
            if(i>4) cow+=add[i-3];
            add[i]=cow;
            ans+=add[i];
        }
        cout<<ans<<endl;
    }
    return 0;
}