题目地址:
https://leetcode.com/problems/best-time-to-buy-and-sell-stock-iii/
给定一个股票的价格,以长 n n n的数组 p p p的形式给出。规定每天只能做一次交易,并且每次只能买 1 1 1股,问如果只允许做最多 2 2 2次买卖,那么最大收益是多少。
思路是前后缀分解。首先预处理出给定前缀 p [ 0 : k ] p[0:k] p[0:k]且只进行一次交易的情况下的最大收益,这个可以用动态规划做。设 f [ k ] f[k] f[k]是上述前缀的最大收益,那么 f [ 0 ] = 0 f[0]=0 f[0]=0, f [ k ] = max { f [ k − 1 ] , p [ k ] − m } , k ≥ 1 f[k]=\max\{f[k-1],p[k]-m\},k\ge 1 f[k]=max{f[k−1],p[k]−m},k≥1, m = min p [ 0 : k − 1 ] m=\min p[0:k-1] m=minp[0:k−1]。算出 f f f之后,可以枚举第二次交易的买入点。设第二次交易的买入点是 u u u,那么总的最大收益是 ( M − p [ u ] ) + f [ u ] (M-p[u])+f[u] (M−p[u])+f[u],其中 M = max p [ u + 1 : n − 1 ] M=\max p[u+1:n-1] M=maxp[u+1:n−1]。这里注意,当第一次卖出点和第二次买入点重合的时候,我们实际上也同时枚举了只交易一次的情况。所以必须得写 ( M − p [ u ] ) + f [ u ] (M-p[u])+f[u] (M−p[u])+f[u]而不能写 ( M − p [ u ] ) + f [ u − 1 ] (M-p[u])+f[u-1] (M−p[u])+f[u−1]。代码如下:
public class Solution {
public int maxProfit(int[] prices) {
if (prices == null || prices.length <= 1) {
return 0;
}
int n = prices.length;
int[] dp = new int[n];
for (int i = 1, min = prices[0]; i < n; i++) {
dp[i] = Math.max(dp[i - 1], prices[i] - min);
min = Math.min(min, prices[i]);
}
int res = 0;
for (int i = n - 1, max = 0; i >= 0; i--) {
res = Math.max(res, max - prices[i] + dp[i]);
max = Math.max(max, prices[i]);
}
return res;
}
}
时空复杂度 O ( n ) O(n) O(n)。