n个台阶,每次可以走一步或者两步,总共有多少种走法。
第一感觉想到的是递归,n为1的时候1种,2的时候2中。其他时候就是 fun(n) = fun(n-1) + fun(n-2);递归的代码很简单。如下
class Solution { public: int climbStairs(int n) { if (n == 0) return 0; if (n == 1) return 1; if (n == 2) return 2; else return climbStairs(n-1) + climbStairs(n-2); } };
但是超时了。看了tag提示DP。于是就分分钟改为DP
class Solution { public: int climbStairs(int n) { if (n == 0) return 0; if (n == 1) return 1; if (n == 2) return 2; vector<int> ans(n); ans[0] = 1; ans[1] = 2; for (int i = 2; i < n; ++i) { ans[i] = ans[i-1] + ans[i-2]; } return ans[n-1]; } };
果断AC。