题目
给定一个数组,它的第 i 个元素是一支给定的股票在第 i 天的价格。
设计一个算法来计算你所能获取的最大利润。你最多可以完成 k 笔交易。
注意: 你不能同时参与多笔交易(你必须在再次购买前出售掉之前的股票)。
示例 1:
输入: [2,4,1], k = 2
输出: 2
解释: 在第 1 天 (股票价格 = 2) 的时候买入,在第 2 天 (股票价格 = 4) 的时候卖出,这笔交易所能获得利润 = 4-2 = 2 。
示例 2:
输入: [3,2,6,5,0,3], k = 2
输出: 7
解释: 在第 2 天 (股票价格 = 2) 的时候买入,在第 3 天 (股票价格 = 6) 的时候卖出, 这笔交易所能获得利润 = 6-2 = 4 。
随后,在第 5 天 (股票价格 = 0) 的时候买入,在第 6 天 (股票价格 = 3) 的时候卖出, 这笔交易所能获得利润 = 3-0 = 3 。
分析
不说了,我只能做出股票1,2题和手续费题,看看人家大神的方法,膜拜就是了。。。
一个通用方法团灭6道股票问题
代码
class Solution {
public:
int maxProfit(int k_max, vector<int>& prices) {
if (prices.empty()){
return 0;
}
int len = prices.size();
if (k_max > len / 2){
return maxProfit_infinity(prices);
}
vector<vector<vector<int>>> dp(len, vector<vector<int>>(k_max + 1, vector<int>(2, 0)));
for (int i = 0; i < len; i++){
for (int k = 1; k <= k_max; k++){
if (i == 0){
dp[i][k][0] = 0;
dp[i][k][1] = -prices[i];
}else{
dp[i][k][0] = max(dp[i - 1][k][0], dp[i - 1][k][1] + prices[i]);
dp[i][k][1] = max(dp[i - 1][k][1], dp[i - 1][k - 1][0] - prices[i]);
}
}
}
return dp[len - 1][k_max][0];
}
int maxProfit_infinity(vector<int>& prices){
int cur_min = INT_MAX, cur_profit = 0, res = 0;
for (auto x : prices){
if (x - cur_min > cur_profit){
cur_profit = x - cur_min;
}else{
res += cur_profit;
cur_profit = 0;
cur_min = x;
}
}
return res + cur_profit;
}
};