力扣 746. 使用最小花费爬楼梯
记爬上n层耗费体力f(n),
爬上n层可分为2种情况:
1)从n-1层爬上n层,f(n)=f(n-1)+cost[n-1]
2) 从n-2层爬上n层,f(n)=f(n-2)+cost[n-2]
则状态转移方程为f(n) = min(f(n-1)+cost[n-1],f(n-2)+cost[n-2])
f(0)=f(1)=0;
class Solution {
public:
int minCostClimbingStairs(vector<int>& cost) {
int tmpN_1=0;
int tmpN_2=0;
int minCost=0;
for(int i=2;i<=cost.size();i++){
minCost = min(tmpN_2+cost[i-2],tmpN_1+cost[i-1]);
tmpN_2 = tmpN_1;
tmpN_1 = minCost;
}
return minCost;
}
};
需要遍历整个cost,时间复杂度为O(n)
只需两个零时变量,空间复杂度为O(1)