母牛的故事

题目

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

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

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

Sample Input
2
4
5
0

Sample Output
2
4
6

解题思路

这道题是dp的入门题，之前我是靠找规律写出的，即从第四年开始，第n年牛是数量 sum(n) = sum (n-1) + sum(n-3)，但是我不知道这个规律到底是怎么来的，但是今天突然想通了。

第n年的牛的数量，其实是由上一年的牛的总数，加上上一年可以生小牛的牛（也就是第n年年初可以生小牛的牛）的总数而得。
上一年的牛的总数即为sum (n-1)，小牛需要过3年才可以生牛，所以第n年年初可以生小牛的牛的总数就是3年前牛的总数（只有这些牛在第n年才可以生牛），即为sum (n-3)。

可得第n年牛是数量 sum(n) = sum (n-1) + sum(n-3)。

题解

考虑到有n个数据，我使用了一个数组用于存储某一年牛的总数，再用了另一个数组记录某一年牛的总数是否被存储

#include<iostream>
using namespace std;

int niu[100] = { 0 };	//用于存储某一年牛的总数
bool judge[100] = { 0 };//记录某一年牛的总数是否被存储

int getsum(int n) {
	if (judge[n]) {	//如果第n年有数据则直接返回，不用继续进行递归
		return niu[n];
	}
	else {	//第n年没有数据
		judge[n] = 1;
		niu[n] = getsum(n - 1) + getsum(n - 3);
		return niu[n];
	}
}

int main() {
	//初始化niu和judge
	niu[1] = 1; niu[2] = 2; niu[3] = 3;
	judge[1] = 1; judge[2] = 1; judge[3] = 1;

	int n;
	while (cin >> n && n != 0) {
		cout << getsum(n) << endl;
	}
	return 0;
}