您的位置:首页 > 大数据 > 人工智能

[LeetCode]746. Min Cost Climbing Stairs

2018-01-11 11:26 344 查看

[LeetCode]746. Min Cost Climbing Stairs

题目描述



思路

不能存在连续两个不选择的数,所以每次更新的公式为

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

考虑选择当前位置,或者选择前一位置的最优结果

代码

#include<iostream>
#include<vector>
#include<algorithm>

using namespace std;

class Solution {
public:
int minCostClimbingStairs(vector<int>& cost) {
if (cost.size() == 2) return min(cost[0], cost[1]);
if (cost.size() == 3) return min(cost[0] + cost[2], cost[1]);
vector<int> dp(cost.size());
dp[0] = cost[0], dp[1] = min(cost[0], cost[1]), dp[2] = min(cost[0] + cost[2], cost[1]);
for (int i = 3; i < cost.size(); ++i)
dp[i] = min(dp[i - 1] + cost[i], dp[i - 2] + cost[i - 1]);
return dp[cost.size() - 1];
}
};

int main() {
vector<int> cost = { 10,15,20,10 };
Solution s;

cout << s.minCostClimbingStairs(cost) << endl;

system("pause");
return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: