题目
给定一个整数数组 prices ,它的第 i 个元素 prices[i] 是一支给定的股票在第 i 天的价格。
设计一个算法来计算你所能获取的最大利润。你最多可以完成 k 笔交易。
注意:你不能同时参与多笔交易(你必须在再次购买前出售掉之前的股票)。
代码
class Solution {
public int maxProfit(int k, int[] prices) {
if (k>prices.length/2)
return maxProfit_inf(prices);
int n=prices.length;
int[][][] dp=new int[n][k+1][2];
// for (int i = 0; i < n; i++) {
// dp[i][0][1]=Integer.MIN_VALUE;//没有操作次数时是不可能持有的
// }
for (int i = 0; i < n; i++) {
for (int j = 1; j<k+1 ; j++) {
if (i==0){
dp[i][j][0]=0;
dp[i][j][1]=-prices[i];
continue;
}
dp[i][j][0]=Math.max(dp[i-1][j][0],dp[i-1][j][1]+prices[i]);
dp[i][j][1]=Math.max(dp[i-1][j][1],dp[i-1][j-1][0]-prices[i]);
}
}
return dp[n-1][k][0];
}
int maxProfit_inf(int[] prices) {
int n =prices.length;
if (n==0) return 0;
int[][] dp=new int[n][2];
for (int i = 0; i < n ; i++) {
if (i==0){
dp[i][0]=0;
dp[i][1]=-prices[i];//base case
continue;
}
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]);
}
return dp[n-1][0];
}
}
要点
- 股票买卖问题,使用有限状态机模板来进行解题
- 根据具体题目要求进行简化以及base 情况的修改即可
- dp表结构: