详见:
https://www.nowcoder.com/practice/9fecec9c776c436b8a03ba0684ac76a7?tpId=225&tags=&title=&difficulty=0&judgeStatus=0&rp=0
实际上是斐波那契数列问题:
f
(
n
)
=
f
(
n
−
1
)
+
f
(
n
−
2
)
f(n) = f(n-1) + f(n-2)
f(n)=f(n−1)+f(n−2)
递归法
时间复杂度:
O
(
2
n
)
O(2^n)
O(2n),如图所示,树型递归,遍历每个结点,最多共
2
n
−
1
2^n-1
2n−1个结点。
空间复杂度:O(n),递归栈最大深度为n。
#include <iostream>
using namespace std;
int getSum(int n);
int main() {
int n;
cin >> n;
cout << getSum(n) << endl;
return 0;
}
int getSum(int n) {
if (n<3) //n=1或2跳出递归 或 if(n == 1 || n == 2)
return 1;
return getSum(n - 1) + getSum(n - 2); //返回前两个月相加
}
迭代法
除了自上而下的递归,我们还可以利用像上述数组一样的自下而上的相加,反正除了第一二月外,这个月的值等于它前两个月的值相加,我们可以从第三个月开始不断累加,并更新前两个月的值即可。
#include <iostream>
using namespace std;
int getSum(int n);
int main() {
int n;
cin >> n;
cout << getSum(n) << endl;
return 0;
}
int getSum(int n) {
if (n<3) //n=1或2跳出递归 或 if(n == 1 || n == 2)
return 1;
int a = 1; //表示当前要计算的月份前一个月
int b = 1; //表示当前要计算月份的前两个月
int res = 0;
for(int i=3; i<=n; i++){ //遍历3-n
res = a + b; //直接相加
//更新前两个
b = a;
a = res;
}
return res;
}
时间复杂度:O(n),一次遍历
空间复杂度:O(1),常数个变量,无额外空间