123.买卖股票的最佳时机III
这道题一下子就难度上来了,关键在于至多买卖两次,这意味着可以买卖一次,可以买卖两次,也可以不买卖。
class Solution {
public int maxProfit(int[] prices) {
// 定义 5 种状态:
// 0: 没有操作, 1: 到现在为止第一次持有 手头上最大现金, 2: 到现在为止第一次不持有(已经卖出过一次了),
// 3: 已经第二次买入了, 4:已经第二次卖出了
int daylen=prices.length;
int[][] dp=new int[daylen][5];
dp[0][0]=0;
dp[0][1]=-prices[0];
dp[0][2]=0;
dp[0][3]=-prices[0];
dp[0][4]=0;
for(int i=1;i<daylen;i++){
// dp[i][0]=0 一直都是0
//第一次持有:原先可以是没持有过,前一天以前也是第一次持有的状态
dp[i][1]=Math.max(dp[i-1][0]-prices[i],dp[i-1][1]);
dp[i][2]=Math.max(dp[i-1][1]+prices[i],dp[i-1][2]);
dp[i][3]=Math.max(dp[i-1][2]-prices[i],dp[i-1][3]);
dp[i][4]=Math.max(dp[i-1][3]+prices[i],dp[i-1][4]);
}
// 卖出现金已经比买入现金多,第二次卖出包括了第一次卖出的情况
return dp[daylen-1][4];
}
}
188.买卖股票的最佳时机IV
本题是123.买卖股票的最佳时机III 的进阶版
class Solution {
public int maxProfit(int k, int[] prices) {
int daylen=prices.length;
//买卖k次 每次有两个状态 再加上0状态 2k+1个状态
int[][] dp=new int[daylen][2*k+1];
dp[0][0]=0;
//奇数项是要第几次买入
for(int i=1;i<2*k+1;i+=2){
dp[0][i]=-prices[0];
}
for(int i=1;i<daylen;i++){
for(int j=0;j<2*k-1;j+=2){
//第x次买或者卖
dp[i][j+1]=Math.max(dp[i-1][j]-prices[i],dp[i-1][j+1]);
dp[i][j+2]=Math.max(dp[i-1][j+1]+prices[i],dp[i-1][j+2]);
}
}
return dp[daylen-1][2*k];
}
}