题目链接

爬楼梯

题目描述

image.png

实现代码

思路:首先可以知道,每一步都是有两个选择:走一步或者走两步,因此,可以采取dfs的方式进行递归;
实现代码:

  1. class Solution {
  2. int result = 0;
  3. public int climbStairs(int n) {
  4. if(n <= 2) {
  5. return n;
  6. }
  7. dfs(n);
  8. return result;
  9. }
  10. public void dfs(int n) {
  11. if(n < 0) {
  12. return;
  13. }
  14. if(n <= 1) {
  15. result++;
  16. return;
  17. }
  18. dfs(n-1);
  19. dfs(n-2);
  20. }
  21. }

结果是正常的,但是运行超时了;

DP思路:记dp[i]表示当楼梯长度为i时的方法数量;可以知道dp[i]与dp[i-1]和dp[i-2]有关,因为只有这两个位置能够到达这里,因此状态转移方程为:

dp[i] = dp[i-1] + dp[i-2];

实现代码为:

  1. class Solution {
  2. int result = 0;
  3. public int climbStairs(int n) {
  4. if(n <= 2) {
  5. return n;
  6. }
  7. int[] dp = new int[n+1];
  8. dp[0] = 0;
  9. dp[1] = 1;
  10. dp[2] = 2;
  11. for(int i=3; i <= n; i++) {
  12. dp[i] = dp[i-1] + dp[i-2];
  13. }
  14. return dp[n];
  15. }
  16. }