123
题目:
给定一个数组,它的第 i
个元素是一支给定的股票在第 i
天的价格。设计一个算法来计算你所能获取的最大利润。你最多可以完成 两笔 交易。注意:你不能同时参与多笔交易(你必须在再次购买前出售掉之前的股票)。
题目链接:123. 买卖股票的最佳时机 III - 力扣(LeetCode)
思路:
动态规划,区别于前面题,这次有5个状态,还要知道dp[prices.length-1][4]肯定是最大值,因为他包括dp[prices.length-1][2],如果第一次卖出已经是最大值了,那么我们可以在当天立刻买入再立刻卖出。所以dp[4][4]已经包含了dp[4][2]的情况。也就是说第二次卖出手里所剩的钱一定是最多的。
代码:
class Solution {
public int maxProfit(int[] prices) {
int[][] dp =new int[prices.length][5];
//0 无买卖 1 买了一次没卖 2 买了一次也卖了 3 买了两次卖了一次 4买了两次都卖了
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<prices.length;i++)
{
dp[i][0]=dp[i-1][0];
dp[i][1] = Math.max(dp[i-1][1],dp[i-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 dp[prices.length-1][4];
}
}
188
题目:
给你一个整数数组 prices
和一个整数 k
,其中 prices[i]
是某支给定的股票在第 i
天的价格。设计一个算法来计算你所能获取的最大利润。你最多可以完成 k
笔交易。也就是说,你最多可以买 k
次,卖 k
次。注意:你不能同时参与多笔交易(你必须在再次购买前出售掉之前的股票)。
题目链接:188. 买卖股票的最佳时机 IV - 力扣(LeetCode)
思路:
于lc123思路相同,只需要把次数2抽象为次数k即可
代码:
class Solution {
public int maxProfit(int k, int[] prices) {
int[][] dp =new int[prices.length][(2*k)+1];
for(int i=0;i<(2*k)+1;i++)
{
if(i%2==0) dp[0][i]=0;
if(i%2==1) dp[0][i]=-prices[0];
}
for(int i=1;i<prices.length;i++)
{
for(int j=0;j<(2*k)+1;j++)
{
if(j==0) dp[i][j]=dp[i-1][j];
else if(j%2==1) dp[i][j] = Math.max(dp[i-1][j],dp[i-1][j-1]-prices[i]);
else if(j%2==0) dp[i][j] = Math.max(dp[i-1][j],dp[i-1][j-1]+prices[i]);
}
}
return dp[prices.length-1][2*k];
}
}