1.题目描述:
给定一个整数数组prices,它的第i个元素prices[i]是一支给定的股票在第i天的价格。设计一个算法来计算你所能获取的最大利润。你最多可以完成k笔交易。注意:你不能同时参与多笔交易(你必须在再次购买前出售掉之前的股票)。
2.动态规划:
做法和leetcode123. 买卖股票的最佳时机 III一致,抽取出来特征即可。
class Solution {
public int maxProfit(int k, int[] prices) {
if (prices.length == 0 || prices.length == 1) return 0;
int[][] dp = new int[prices.length][2 * k + 1];
for (int i = 1; i < 2 * k + 1; i += 2) {
dp[0][i] = -prices[0];//奇数次全是买入操作
}
for (int i = 1; i < prices.length; i++) {
for (int j = 1; j < 2 * k + 1; j++) {
if (j % 2 == 1) dp[i][j] = Math.max(dp[i - 1][j], dp[i - 1][j - 1] - prices[i]);//买入操作
else dp[i][j] = Math.max(dp[i - 1][j], dp[i - 1][j - 1] + prices[i]);//卖出操作
}
}
return dp[dp.length - 1][2 * k];
}
}
二刷:
class Solution {
public int maxProfit(int k, int[] prices) {
if (prices.length == 0 || prices.length == 1) return 0;
int[][] dp = new int[prices.length][2 * k + 1];
for (int i = 0; i < k; i++) {
dp[0][2 * i + 1] = -prices[0];//买入
}
for (int i = 1; i < dp.length; i++) {
dp[i][0] = dp[i - 1][0];
for (int j = 0; j < k; j++) {
dp[i][2 * j + 1] = Math.max(dp[i - 1][2 * j + 1], dp[i - 1][2 * j] - prices[i]);//第j + 1次买入
dp[i][2 * j + 2] = Math.max(dp[i - 1][2 * j + 2], dp[i - 1][2 * j + 1] + prices[i]);//第j + 1次卖出
}
}
return dp[dp.length - 1][2 * k];
}
}