代码随想录算法训练营第五十一天|309.最佳买卖股票的时机含冷冻期、714.买卖股票的最佳时机函手续费
309.最佳买卖股票的时机含冷冻期
给定一个整数数组prices
,其中第 prices[i]
表示第 *i*
天的股票价格 。
设计一个算法计算出最大利润。在满足以下约束条件下,你可以尽可能地完成更多的交易(多次买卖一支股票):
- 卖出股票后,你无法在第二天买入股票 (即冷冻期为 1 天)。
**注意:**你不能同时参与多笔交易(你必须在再次购买前出售掉之前的股票)。
示例 1:
输入: prices = [1,2,3,0,2]
输出: 3
解释: 对应的交易状态为: [买入, 卖出, 冷冻期, 买入, 卖出]
题解:在买卖股票II的基础上增加了一个冷冻期,冷冻期是在卖出股票的后一天,所以要将不持有股票的状态分开来,分成保持卖出状态和卖出状态,这样才能找到冷冻期,也就是本题有4个状态。
- dp(i,0) :持有股票
dp(i,1) :保持卖出股票
dp(i,2) :卖出股票
dp(i,3) :冷冻期
- 递推公式:
dp(i,0)=Math.max(dp(i-1,0),dp(i-1,1)-prices[i] ,dp(i,3)-prices[i])
dp(i,1)=Math.max(dp(i-1,1),dp(i-1,3))
dp(i,2)=(dp(i-1,0)+prices[i])
dp(i,3)=dp(i-1,3)
- 初始化
- 遍历顺序:从前到后
- 打印dp
代码:
class Solution {
public int maxProfit(int[] prices) {
int len=prices.length;
int [][]dp=new int [len][4];
//持股
dp[0][0]=-prices[0];
//保持卖出
dp[0][1]=0;
//卖出
dp[0][2]=0;
//冷冻期
dp[0][3]=0;
for(int i=1;i<len;i++){
dp[i][0]=Math.max(dp[i-1][0],Math.max(dp[i-1][1]-prices[i],dp[i-1][3]-prices[i]));
dp[i][1]=Math.max(dp[i-1][1],dp[i-1][3]);
dp[i][2]=dp[i-1][0]+prices[i];
dp[i][3]=dp[i-1][2];
}
int max=Integer.MIN_VALUE;
for(int n:dp[len-1]){
max=Math.max(max,n);
}
return max;
}
}
714.买卖股票的最佳时机函手续费
给定一个整数数组 prices
,其中 prices[i]
表示第 i
天的股票价格 ;整数 fee
代表了交易股票的手续费用。
你可以无限次地完成交易,但是你每笔交易都需要付手续费。如果你已经购买了一个股票,在卖出它之前你就不能再继续购买股票了。
返回获得利润的最大值。
**注意:**这里的一笔交易指买入持有并卖出股票的整个过程,每笔交易你只需要为支付一次手续费。
示例 1:
输入:prices = [1, 3, 2, 8, 4, 9], fee = 2
输出:8
解释:能够达到的最大利润:
在此处买入 prices[0] = 1
在此处卖出 prices[3] = 8
在此处买入 prices[4] = 4
在此处卖出 prices[5] = 9
总利润: ((8 - 1) - 2) + ((9 - 4) - 2) = 8
题解:最多只能持有一支股票。嗯?这个题目描述有点模糊啊,每一笔交易都要支付手续费的话不应该是买入要手续费,卖出也要手续费吗,这个通过代码买入和卖出只需要一笔手续费。
dp解题步骤和买卖股票的最佳时机II是一样的,在那个代码的基础上减去手续费即可。
代码:
class Solution {
public int maxProfit(int[] prices, int fee) {
int len=prices.length;
int [][] dp=new int [len][2];
//不持有股票
dp[0][0]=0;
//持有股票
dp[0][1]=-prices[0];
for(int i=1;i<len;i++){
dp[i][0]=Math.max(dp[i-1][0],dp[i-1][1]+prices[i]-fee);
dp[i][1]=Math.max(dp[i-1][1],dp[i-1][0]-prices[i]);
}
return Math.max(dp[len-1][0],dp[len-1][1]);
}
}