动态规划之股票问题
文章目录
1. “121. 买卖股票的最佳时机”
](https://i0.hdslb.com/bfs/album/a10d1c5b8ea00e9379a0b2b8439d2330b62962a6.png)
首先确定dp数组的含义,dp[i][0]代表第i天不持有股票所剩余金额,dp[i][1]代表第i天持有股票所剩余金额。
那么第i天不持有股票可以是前一天就不持有股票,即dp[i-1][0],也可以是前一天持有股票,第i天卖出股票即dp[i-1][1]+prices[i],两者取最大值,列出状态转移方程dp[i][0]=Math.max(dp[i-1][0],dp[i-1][1]+prices[i])。
同样的,第i天持有股票可以是前一天就持有股票,即dp[i-1][1],也可以是前一天不持有股票,第i天买入股票即dp[i-1][0]-prices[i],两者取最大值,列出状态转移方程dp[i][1]=Math.max(dp[i-1][1],dp[i-1][0]-prices[i])。
因此得出
dp[i][0]=Math.max(dp[i-1][0],dp[i-1][1]+prices[i])
dp[i][1]=Math.max(dp[i-1][1],dp[i-1][0]-prices[i])
然后因为本题因为只能买入一次,买入股票后所剩余金额必定等于0-股票价格,因此dp[i][1]简化为dp[i][1]=Math.max(dp[i-1][1],-prices[i])
最后,因为取值只依赖于dp[0]和dp[1],简化为一维数组
class Solution {
public int maxProfit(int[] prices) {
int[] dp = new int[2];
dp[0] = 0;
dp[1] = 0 - prices[0];
for (int i = 1; i < prices.length; i++) {
dp[0] = Math.max(dp[0], dp[1] + prices[i]);
dp[1] = Math.max(dp[1], 0 - prices[i]);
}
return dp[0];
}
}
2. “122. 买卖股票的最佳时机 II”
整体思路与上一题一致,不过不同的是这题可以无限次的买卖同一只股票,这就意味着你买入股票剩余的金额不再是-prices[i]了,而应该是前一天不持有股票所剩金额-prices[i]
即dp[i][1]=Math.max(dp[i-1][1],dp[i-1][0]-prices[i])
而di[i][0]第i天不持有股票和上一题一致dp[i][0]=Math.max(dp[i-1][0],dp[i-1][1]+prices[i])
class Solution {
public int maxProfit(int[] prices) {
int[] dp = new int[2];
dp[0] = 0;
dp[1] = 0 - prices[0];
for (int i = 1; i < prices.length; i++) {
dp[0] = Math.max(dp[0], dp[1] + prices[i]);
dp[1] = Math.max(dp[1], dp[0] - prices[i]);
}
return dp[0];
}
}
3. “123. 买卖股票的最佳时机III”
这一题与上两题不同的是交易次数限制为两次,因此设置一个未知数k代表交易次数,并且dp数组可以些为dp[i][k][0]和dp[i][k][1]代表最多能有k次交易
并且一次交易代表着买入和卖出两个操作,因此我们只需在买入时将k-1就可以了,所以状态转移方程为
dp[i][k][0] = Math.max(dp[i - 1][k][0], dp[i - 1][k][1] + prices[i])
dp[i][k][1] = Math.max(dp[i - 1][k][1], dp[i - 1][k - 1][0] - prices[i])
本题交易次数限制为两次,将k=2带入得到
dp[i][2][0] = Math.max(dp[i - 1][2][0], dp[i - 1][2][1] + prices[i])
dp[i][2][1] = Math.max(dp[i - 1][2][1], dp[i - 1][1][0] - prices[i])
dp[i][1][0] = Math.max(dp[i - 1][1][0], dp[i - 1][1][1] + prices[i])
dp[i][1][1] = Math.max(dp[i - 1][1][1], dp[i - 1][0][0] - prices[i])
= Math.max(dp[i - 1][1][1], -prices[i])
简化一下就是
dp_i20 = Math.max(dp_i20, dp_i21 + prices[i])
dp_i21 = Math.max(dp_i21, dp_i10 - prices[i])
dp_i10 = Math.max(dp_i10, dp_i11 + prices[i])
dp_i11 = Math.max(dp_i11, -prices[i])
然后考虑他们的初值,第0天不持有,剩余金额肯定为0,因此dp_i20和dp_i10初值为0
第0天持有,剩余金额肯定为-prices[0],因此dp_i21和dpi11初值为-prices[0]
class Solution {
public int maxProfit(int[] prices) {
int i02 = 0;
int i12 = 0 - prices[0];
int i01 = 0;
int i11 = 0 - prices[0];
for (int i = 0; i < prices.length; i++) {
i02 = Math.max(i02, i12 + prices[i]);
i12 = Math.max(i12, i01 - prices[i]);
i01 = Math.max(i01, i11 + prices[i]);
i11 = Math.max(i11, -prices[i]);
}
return i02;
}
}
4. “188. 买卖股票的最佳时机 IV”
这题的条件改为了交易次数最多为k次,求最大利润。思路和上一题是一模一样的,只不过上一次k=2可以把情况穷举出来,而这题k是未知数,因此使用三维数组。
class Solution {
public int maxProfit(int k, int[] prices) {
if (prices.length == 0) {
return 0;
}
k = Math.min(k, prices.length / 2);
int[][][] dp = new int[prices.length][2][k + 1];
for (int i = 0; i <= k; i++) {
dp[0][0][i] = 0;
dp[0][1][i] = -prices[0];
}
for (int i = 1; i < prices.length; i++) {
for (int j = k; j > 0; j--) {
dp[i][0][j] = Math.max(dp[i - 1][0][j], dp[i - 1][1][j] + prices[i]);
dp[i][1][j] = Math.max(dp[i - 1][1][j], dp[i - 1][0][j - 1] - prices[i]);
}
}
return dp[prices.length - 1][0][k];
}
}
不过要注意交易次数最大为prices[]的长度除以2才是有效的,因为一次买入必须对应着一次卖出,交易次数最多为prices[]的长度除以2。
5. “309. 最佳买卖股票时机含冷冻期”
这一题的条件为卖出股票后无法在第二天买入股票,也就是说在第i天卖出股票,最少得在第i+2天才能买入,
也就是说dp[i][1]应该为dp[i][1]=Math.max(dp[i-1][1],dp[i-2][0]-prices[i]),但是当i=1时,也就是在第1天持有股票时,股票肯定是在第0天或者第1天买入的
然后并没有规定买入股票有冷冻期,因此买入股票可以在第二天卖出,dp[i][0]不变还是dp[i][0]=Math.max(dp[i-1][0],dp[i-1][1]+prices[i])
class Solution {
public int maxProfit(int[] prices) {
int[][] dp = new int[prices.length][2];
dp[0][0] = 0;
dp[0][1] = -prices[0];
for (int i = 1; i < prices.length; i++) {
dp[i][0] = Math.max(dp[i - 1][0], dp[i - 1][1] + prices[i]);
dp[i][1] = i > 1 ? Math.max(dp[i - 1][1], dp[i - 2][0] - prices[i]) : Math.max(dp[i - 1][1], dp[i - 1][0] - prices[i]);
}
return dp[prices.length - 1][0];
}
}
6. “714. 买卖股票的最佳时机含手续费”
这题的条件加了一个手续费,每笔交易都需支付一个固定的手续费,我们只需在卖出时减去手续费就行了,还是熟悉的套路
class Solution {
public int maxProfit(int[] prices, int fee) {
int[] dp = new int[2];
dp[0] = 0;
dp[1] = -prices[0];
for (int i = 1; i < prices.length; i++) {
dp[0] = Math.max(dp[0], dp[1] + prices[i] - fee);
dp[1] = Math.max(dp[1], dp[0] - prices[i]);
}
return dp[0];
}
}