算法题目：母牛的故事

问题 1004: 母牛的故事

时间限制: 1Sec 内存限制: 128MB 提交: 20267 解决: 5615

题目描述

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

输入

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

输出

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

样例输入

2
4
5
0

样例输出

2
4
6

 

http://www.dotcpp.com/oj/problem1004.html

 

题解

#include <iostream>

using namespace std;

long f(int n){
	if(n==1||n==2||n==3){
		return n;
	}
	long num;
	long n1 = 1;
	long n2 = 2;
	long n3 = 3;
	for(int i=4;i<=n;i++){
		num = n3 + n1;
		n1 = n2;
		n2 = n3;
		n3 = num;
	}
	return num;
}

int main(){
	int n;
	while(cin>>n){
		if(n==0){
			break;
		}
		cout<<f(n)<<endl;
	}
	return 0;
} 

跟斐波那契很相似

这个题目如过用递归会超时.

思路

年：1     2       3     4    5     6   7 

头：1     2       3     4    6     9   13

可以看到当n>=4

f(n) = f(n-1) + f(n-3)

所以循环的时候就应该把

n1 = 1      n2 = 2     n3 = 3      num = 4 = n3+n1

变成

n1 = 2      n2 = 3      n3 = 4     num

相当于把n1 n2 n3 向左平移  把n1扔掉

num = n3 + n1

n1 = n2

n2 = n3

n3 = num