123. 买卖股票的最佳时机 III
难题,关键在于状态的转换,很麻烦,本题中用到了五种状态表示每一次的买卖股票
dp[0]:没有买卖过
dp[1]:第一次持有
dp[2]:第一次卖出
dp[3]:第二次持有
dp[4]:第二次卖出
通过这个状态转换即可很好做题,代码如下:
public int maxProfit(int[] prices) {
int[][] dp = new int[prices.length][5];
dp[0][0] = 0;
dp[0][1] = -prices[0];
dp[0][3] = -prices[0];
for(int i=1;i<prices.length;i++){
dp[i][1] = Math.max(dp[i-1][1],0-prices[i]);
dp[i][2] = Math.max(dp[i-1][2],dp[i-1][1]+prices[i]);
dp[i][3] = Math.max(dp[i-1][3],dp[i-1][2]-prices[i]);
dp[i][4] = Math.max(dp[i-1][4],dp[i-1][3]+prices[i]);
}
return Math.max(Math.max(dp[dp.length-1][1],dp[dp.length-1][2]),Math.max(dp[dp.length-1][3],dp[dp.length-1][4]));
}
188. 买卖股票的最佳时机 IV
也是难题,此次的状态有2*k+1个,需要用三个for循环来完成代码的实现,总的来说就是将最多可以卖几次进行集成化,代码如下:
public int maxProfit(int k, int[] prices) {
int[][] dp = new int[prices.length][k*2+1];
for(int i=1;i<k*2+1;i++){
dp[0][i] = -prices[0];
i++;
}
for(int i=1;i<prices.length;i++){
for(int j=1;j<k*2+1;j++){
dp[i][j] = Math.max(dp[i-1][j],dp[i-1][j-1]-prices[i]);
j++;
dp[i][j] = Math.max(dp[i-1][j],dp[i-1][j-1]+prices[i]);
}
}
int re = 0;
for(int i=0;i<k*2+1;i++){
if(dp[dp.length-1][i]>re){
re = dp[dp.length-1][i];
}
}
return re;
}