图片. PNG
基本思想: 用 dp[n] 大小为 n 的数组保存 sum 结果. 最后返回的时候判断 min(dp[n-1],dp[n-2]).
解释: 最后返回的时候可以从最后一个返回, 也可以从倒数第二个返回.
图片. PNG
C++:
- int minCostClimbingStairs(vector<int>& cost) {
- if(cost.size()==0)
- return 0;
- int n=cost.size();
- vector<int>dp(n,0);
- if(n==2)
- return min(cost[0],cost[1]);
- if(n==1)
- return dp[0];
- dp[0]=cost[0];dp[1]=cost[1];
- for(int i=2;i<n;i++){
- dp[i]=min(dp[i-1],dp[i-2])+cost[i];
- }
- return min(dp[n-1],dp[n-2]);
- }
Java:
- //faster
- public int minCostClimbingStairs(int[] cost) {
- int n = cost.length;
- int[] dp = new int[n];
- dp[0] = cost[0];
- dp[1] = cost[1];
- for (int i = 2; i < n; i++) {
- dp[i] = cost[i] + Math.min(dp[i - 1], dp[i - 2]);
- }
- return Math.min(dp[n - 1], dp[n - 2]);
- }
来源: http://www.jianshu.com/p/16b4aacc9426