- 可以從下標0或者1作為起始位置————dp[0] = dp[1] = 0。
- 一次性可以選擇移動1次或者2次,故當下標>=2的時候,到達2有可能是從下標0開始或者下標1開始,cost[0] or cost[1];到達n,有可能是花費cost[n-1]到達,也有可能花費cost[n-2]到達。取最小值。
class Solution {public int minCostClimbingStairs(int[] cost) {int n = cost.length;int[] dp = new int[n+1];dp[0]=dp[1]=0;for(int i=2;i<=n;i++){dp[i] = Math.min(dp[i-1]+cost[i-1], dp[i-2]+cost[i-2]);}return dp[n];}
}
優化
看到當前的結果計算的時候,只需要上一個和上兩個的值,所有使用“滾動數組”的思路。
class Solution {public int minCostClimbingStairs(int[] cost) {int n = cost.length;int[] dp = new int[n+1];dp[0]=dp[1]=0;int dp1 = 0;int dp2 = 0;for(int i=2;i<=n;i++){dp[i] = Math.min(dp2+cost[i-1], dp1+cost[i-2]);dp1 = dp[i-1];dp2 = dp[i];}return dp[n];}
}