程序设计基础II-实验3-递推算法设计 7-3 母牛的故事

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

输入格式:

输入为一个整数n(0< n< 55)

输出格式:

输出在第n年的时候母牛的数量。

输入样例1:

2

输出样例1:

2

输入样例2:

5

输出样例2:

6

参考代码: 

#include <stdio.h>

long long num(int n) {
	long long a[55] = {0,1,2,3,4};
	for ( int i = 5; i <= n; i++)
		a[i] = a[i-1] + a[i-3];
	return a[n];
}

int main() {
	int n;
	scanf("%d", &n);
	printf("%lld\n", num(n));
	return 0;
}