我一开始,写了一个代码
class Solution {
public:
int climbStairs(int n) {
vector<int> dp(n+1);
dp[1]=1;
dp[2]=2;
if(n==1){
return 1;
}else if(n==2){
return 2;
}else{
for(int i=3;i<=n;i++){
dp[i]=dp[i-1]+dp[i-2];
}
return dp[n];
}
}
};
一直报错,找了半天终于找到了当n为1时,数组只有2个int型内存,然而后面的
dp[1]=1;
dp[2]=2;
在索引为2时,有越界了,所以改的话:
class Solution {
public:
int climbStairs(int n) {
vector<int> dp(n+5);
dp[1]=1;
dp[2]=2;
if(n==1){
return 1;
}else if(n==2){
return 2;
}else{
for(int i=3;i<=n;i++){
dp[i]=dp[i-1]+dp[i-2];
}
return dp[n];
}
}
};
扩大一下就行了。