122. Best Time to Buy and Sell Stock II


DP
time: O(n)
space:O(n)
class Solution {
public int maxProfit(int[] prices) {
int n = prices.length;
int dp[][] = new int[n][2];
dp[0][0] = 0;
dp[0][1] = -prices[0];
for (int i = 1; i < n ; i++) {
dp[i][0] = Math.max(dp[i-1][0], dp[i-1][1] + prices[i]);
dp[i][1] = Math.max(dp[i-1][1], dp[i-1][0] - prices[i]);
}
return dp[n-1][0];
}
}
use variable
注意到上面的狀態轉移方程中,每一天的狀態只與前一天的狀態有關,而與更早的狀態都無關,因此我們不必存儲這些無關的狀態,
只需要將 dp[i-1] [0] , dp[i-1] [1] 存放在兩個變量中,通過它們計算出
dp[i] [0], dp [ i ] [ 1 ] 並存回對應的變量
class Solution {
public int maxProfit(int[] prices) {
int n = prices.length;
int dp0 = 0;
int dp1 = -prices[0];
for (int i = 1; i < n ; i++) {
int newDp0 = Math.max(dp0, dp1 + prices[i]);
int newDp1 = Math.max(dp1, dp0 - prices[i]);
dp0 = newDp0;
dp1 = newDp1;
}
return dp0;
}
}
Last updated
Was this helpful?