309.最佳买卖股票时机含冷冻期
这题我一开始的想法是,冷冻期代表着买入股票时受限,需要从i-2天的not hold来考虑购买。
class Solution {
public int maxProfit(int[] prices) {
int[][] dp = new int[prices.length][2];
if(prices.length == 1) return 0;
dp[0][0] = -prices[0];
dp[0][1] = 0;
dp[1][0] = Math.max(-prices[0], -prices[1]);
dp[1][1] = Math.max(dp[0][0]+prices[1], 0);
for (int i = 2; i < prices.length; i++ ){
// 0 hold, 1 not hold
dp[i][0] = Math.max(dp[i-1][0], dp[i-2][1]-prices[i]);
dp[i][1] = Math.max(dp[i-1][0] + prices[i], dp[i-1][1]);
}
return dp[dp.length-1][1];
}
}
这是细分了状态转移,尤其是not hold stock时的,刚卖出,冷冻期,持续not hold
具体可以区分出如下四个状态:
状态一:持有股票状态(今天买入股票,或者是之前就买入了股票然后没有操作,一直持有)
不持有股票状态
状态二:保持卖出股票的状态(两天前就卖出了股票,度过一天冷冻期。或者是前一天就是卖出股票状态,一直没操作)
状态三:今天卖出股票
状态四:今天为冷冻期状态,但冷冻期状态不可持续,只有一天!
class Solution {
public int maxProfit(int[] prices) {
// 0: hold stock
// 1: sold today
// 2: cooldown
// 3: not hold stock and no other status
int[][] dp = new int[prices.length][4];
dp[0][0] = -prices[0];
for (int i = 1; i < prices.length; i++) {
dp[i][0] = Math.max(dp[i-1][0], Math.max(dp[i-1][2]-prices[i], dp[i-1][3]-prices[i]));
dp[i][1] = dp[i-1][0]+prices[i];
dp[i][2] = dp[i-1][1];
dp[i][3] = Math.max(dp[i-1][2], dp[i-1][3]);
}
return Math.max(dp[dp.length-1][1], Math.max(dp[dp.length-1][2], dp[dp.length-1][3]));
}
}
时间复杂度:O(n)
空间复杂度:O(n)
● 714.买卖股票的最佳时机含手续费
与ii和相似,只是多了一步需要在卖出时,减去手续费
class Solution {
public int maxProfit(int[] prices, int fee) {
int[][] dp = new int[prices.length][2];
dp[0][0] = -prices[0];
dp[0][1] = 0;
for (int i = 1; i < prices.length; i++) {
dp[i][0] = Math.max(dp[i-1][0], dp[i-1][1]-prices[i]); // hold
dp[i][1] = Math.max(dp[i-1][0]+prices[i]-fee, dp[i-1][1]); // not hold
}
return dp[dp.length-1][1];
}
}
时间复杂度:O(n)
空间复杂度:O(n)