题目描述
有一种兔子,从出生后第3个月起每个月都生一只兔子,小兔子长到第三个月后每个月又生一只兔子。
例子:假设一只兔子第3个月出生,那么它第5个月开始会每个月生一只兔子。
一月的时候有一只兔子,假如兔子都不死,问第n个月的兔子总数为多少?
数据范围:输入满足 1≤n≤31
题目来源
[题目来源]统计每个月兔子的总数_牛客题霸_牛客网 (nowcoder.com)
输入描述
输入一个int型整数表示第n个月
输出描述:
输出对应的兔子总数
输入:
3
输出:
2
思路
【递归】对于f(n)=f(n−1)+f(n−2))的问题,我们可以每次递归地寻找子问题,然后子问题累得到结果,子问题找到1或者2就停止。
具体实现
#include<iostream>
using namespace std;
//求第n个月兔子出生数量
int getsum(int n)
{
if (n == 1 || n == 2)
return 1;//n=1或n=2跳出递归
return getsum(n - 1) + getsum(n - 2);//
}
int main()
{
int n;
while (cin >> n)
{
cout << getsum(n) << endl;
}
system("pause");
return 0;
}
时间复杂度
- 时间复杂度:O(2的n次方),如图所示,树型递归,遍历每个结点
- 空间复杂度:O(n),递归栈最大深度为n
小结
-
对于递推关系式,递归解决问题简单粗暴
-
参考另一种做法,非递归方式。时间复杂度:O(n),空间复杂度:O(1)
#include<iostream>
using namespace std;
int main(){
int n;
while(cin >> n){ //每次输入n
if(n <= 2) //前两个月直接输出
cout << 1 << endl;
else{
int dpi_2 = 1; //初始化第1个月
int dpi_1 = 1; //初始化第2个月
int output = 0;
for(int i = 3; i <= n; i++){
output = dpi_1 + dpi_2; //公式相加
dpi_2 = dpi_1; //变量更新
dpi_1 = output;
}
cout << output << endl;
}
}
return 0;
}