天天看点

牛客网:最小花费爬楼梯

牛客网:最小花费爬楼梯
牛客网:最小花费爬楼梯

 这个很明显要用动态规划来做,首先,我们设一个动态规划的数组dp[n+1],其中n是楼梯数,dp[i]表示到达第i个台阶的时候最小的花费。设置一个cost[n]来存储爬某一个楼梯要用的代价。第一个楼梯的代价是cost[n-1],以此类推。

接下来列出动态转移方程,很明显,我们要么在之前跨了两级,要么只跨了一级。如果是跨了两级。

dp[i]=min(dp[i-2]+cost[i-2],dp[i-1]+cost[i-1])

代码如下所示:

#include<bits/stdc++.h>
using namespace std;
int main(){
    int n;
    cin>>n;
    vector<int> cost(n);
    for(int i=0;i<n;++i){
        cin>>cost[i];
    }
    vector<int> dp(n+1);
    for(int i=2;i<=n;++i){
        dp[i]=min(dp[i-2]+cost[i-2], dp[i-1]+cost[i-1]);
    }
    cout<<dp[n]<<endl;
    return 0;
}
           

继续阅读