题目描述
一只青蛙一次可以跳上1级台阶,也可以跳上2级。求该青蛙跳上一个n级的台阶总共有多少种跳法(先后次序不同算不同的结果)。
仔细分析,这是一道经典的递归问题,也可以用动态规划求解
解法一:动态规划
用动态规划的思想,则我们从小问题开始求解,并用数组res
存储每个问题的解决方案个数,数组从1开始:
- 假设有1级台阶,那么求解方案只有一种,即只跳1级,则
res[1] = 1
- 假设有2级台阶,那么解决方案有2两种,即一次性跳2级或分两次每次跳1级,则
res[2] = 2
- 假设有3级台阶,由于每次最多跳2级,则3级台阶的前一次(即一次性跳到第3级)可以从2级台阶往上跳1级 或 1级台阶往上跳2级,因此3级台阶的跳法个数就是2级和1级跳法个数的和,即
res[3] = res[2] + res[1]
因此可以得到每个问题间的解决方案关系,res[n] = res[n-1] + res[n-2]
,java实现如下
public class Solution {
public int JumpFloor(int target) {
if (target == 1)
return 1;
if (target == 2)
return 2;
// DP
int res[] = new int[target + 1];
res[1] = 1;
res[2] = 2;
for (int i = 3; i <= target; i++)
res[i] = res[i - 1] + res[i - 2];
return res[target];
}
}
解法二:递归(带备忘录)
能用DP当然也能用递归,递归就是先从大问题来考虑,要跳到第n级台阶,那上一次肯定是从n-1级或者是n-2级台阶跳上来的,因此只需要知道n-1和n-2级台阶的答案就能知道n级台阶的答案了,递推公式如下:
f
(
n
)
=
f
(
n
−
1
)
+
f
(
n
−
2
)
,
n
>
2
f(n) = f(n-1) + f(n-2), n>2
f(n)=f(n−1)+f(n−2),n>2
递归出口为n=1或n=2时,实现如下
public class Solution {
public int res[]; // 备忘录
public int JumpFloor(int target) {
res = new int[target + 1];
for (int i = 1; i <= target; i++)
res[i] = -1;
return f(target);
}
public int f(int n) {
if (n == 1)
res[n] = 1;
if (n == 2)
res[n] = 2;
if (res[n] != -1)
return res[n]; // 如果已经解决过,就直接返回答案
int res_1 = f(n - 1);
int res_2 = f(n - 2);
res[n] = res_1 + res_2;
return res[n];
}
}