今天学习了动态规划,其本质是将一个大问题拆成几个子问题,分别求解这些子问题,从而推断出大问题的解。思想上有一些类似递归,但时间复杂度上比递归更胜一筹。以下是第一道动态规划题:
给出一个正整数n,为当前需要爬的楼梯数,我们可以爬一节或者爬两节,求一共有多少种不同的方法可以爬上楼梯。这里我们设f(n) 为不同方法的个数,因为每一次我们都可以选择爬一节或者爬两节,所以当前爬n节楼梯总方法数可以表示为爬n-1节楼梯方法数和爬n-2节楼梯方法数的和,表示为:
由此,通过动态规划,我们可以写出以下代码:
class Solution {
public:
int climbStairs(int n) {
if(n == 1) return 1; //排除只需要爬一层的情况,不用考虑两层的原因是dp[0]默认为0
vector<int> dp(n+1); //建立动态规划数组
dp[1] = 1; //爬一层时方法数为1
dp[2] = 2; //爬两层时方法数为2
for(int i = 3; i<= n; ++i) //对层数循环,并记录每一层的方法数
{
dp[i] = dp[i-1] + dp[i-2];
}
return dp[n];
}
};