给定一个整数数组 prices
,它的第 i
个元素 prices[i]
是一支给定的股票在第 i
天的价格。
设计一个算法来计算你所能获取的最大利润。你最多可以完成 k 笔交易。
注意:你不能同时参与多笔交易(你必须在再次购买前出售掉之前的股票)。
示例 1:
输入:k = 2, prices = [2,4,1] 输出:2 解释:在第 1 天 (股票价格 = 2) 的时候买入,在第 2 天 (股票价格 = 4) 的时候卖出,这笔交易所能获得利润 = 4-2 = 2 。
示例 2:
输入:k = 2, prices = [3,2,6,5,0,3] 输出:7 解释:在第 2 天 (股票价格 = 2) 的时候买入,在第 3 天 (股票价格 = 6) 的时候卖出, 这笔交易所能获得利润 = 6-2 = 4 。 随后,在第 5 天 (股票价格 = 0) 的时候买入,在第 6 天 (股票价格 = 3) 的时候卖出, 这笔交易所能获得利润 = 3-0 = 3 。
public int maxProfit(int k, int[] prices) {
if (prices.length == 0) {
return 0;
}
if (k >= prices.length / 2) {
return allMaxProfit(prices);
}
int[][] dp = new int[prices.length][k + 1];
for (int i = 1; i < prices.length; i++) {
for (int j = 1; j <= k; j++) {
dp[i][j] = dp[i - 1][j];
for (int p = 0; p < i; p++) {
dp[i][j] = Math.max(dp[i][j - 1] + prices[i] - prices[p], dp[i][j]);
}
}
}
return dp[prices.length - 1][k];
}
public int allMaxProfit(int[] prices) {
int ans = 0;
for (int i = 1; i < prices.length; i++) {
if (prices[i - 1] < prices[i]) {
ans += Math.max(prices[i] - prices[i - 1],0);
}
}
return ans;
}
优化:
public static int maxProfit2(int k, int[] prices) {
if (prices.length == 0) {
return 0;
}
if (k >= prices.length / 2) {
return allMaxProfit(prices);
}
int[][] dp = new int[k+1][prices.length];
int ans = 0;
for (int j = 1; j <= k; j++) {
int pre = dp[j][0];
int best = pre - prices[0];
for (int i = 1; i < prices.length; i++) {
pre = dp[j-1][i];
dp[j][i] = Math.max(dp[j][i - 1], prices[i] + best);
best = Math.max(best, pre - prices[i]);
ans = Math.max(ans, dp[j][i]);
}
}
return ans;
}
public static int allMaxProfit(int[] prices) {
int ans = 0;
for (int i = 1; i < prices.length; i++) {
if (prices[i - 1] < prices[i]) {
ans += Math.max(prices[i] - prices[i - 1], 0);
}
}
return ans;
}