给定数组 prices, 其中第 i 个元素代表某只股票在第 i 天第价格.
你最多可以完成 k 笔交易. 问最大的利润是多少?
思路:这个与Ⅲ是类似的, 就是将只能交易2次,变成了可以交易k次。思路是一样,有2k + 1种情况。然后我们也是分别列举就可以了。需要注意的是,当 k 大于数组长度的一半时,就可以转为普通的不计次数的情况来进行操作。
class Solution {
public int maxProfit(int k, int[] prices) {
int n = prices.length;
if(n == 0 || n == 1)return 0;
if(k > (n / 2)) return maxProfit1(prices);
int[][] f = new int[n + 1][2 * k + 2];
f[0][1] = 0;
for(int i = 2; i <= 2 * k + 1; i++){
f[0][i] = Integer.MIN_VALUE;
}
for(int i = 1; i <= n; i++){
for(int j = 1; j <= 2 * k + 1; j += 2){
f[i][j] = f[i - 1][j];
if(i >= 2 && j > 1 && f[i - 1][j - 1] != Integer.MIN_VALUE){
f[i][j] = Math.max(f[i][j], f[i - 1][j - 1] + prices[i - 1] - prices[i - 2]);
}
}
for(int j = 2; j <= 2 * k + 1; j += 2){
f[i][j] = f[i - 1][j - 1];
if(i >= 2 && f[i - 1][j] != Integer.MIN_VALUE){
f[i][j] = Math.max(f[i][j], f[i - 1][j] + prices[i - 1] - prices[i - 2]);
}
}
}
int res = 0;
for(int j = 1; j <= 2 * k + 1; j += 2){
res = Math.max(res, f[n][j]);
}
return res;
}
public static int maxProfit1(int[] prices){
int n = prices.length;
int res = 0;
for(int i = 1; i < n; i++){
if(prices[i] > prices[i - 1]){
res += prices[i] - prices[i - 1];
}
}
return res;
}
}