188.买卖股票的最佳时机IV
看完想法:是买卖股票Ⅲ的复杂版,有第一次买,第一次卖,.....第N次卖。在遍历的时候,得用一个for循环来定义第二个维度的dp值。当然买是奇数,卖是偶数。初始化的时候,仍然需要一个循环。注意最终程序的时间复杂度是O(n * k)
int maxProfit(int k, vector<int>& prices) {
if(prices.size() == 0) return 0;
vector<vector<int>> dp(prices.size(), vector<int>(2*k+1,0));
//记得初始化
for (int j = 1; j < 2 * k; j += 2) {
dp[0][j] = -prices[0];
}
for(int i = 1; i<prices.size(); i++){
//注意这里j<2k-1,因为for中j+=2,所以需要-1
for(int j = 0; j< 2*k-1; j+=2){
//第i天持有股票,两种情况,和上次保持以及i天买入,注意i, j
dp[i][j+1] = max(dp[i-1][j+1], dp[i-1][j] - prices[i]);
//第i天不持有股票,两种情况,和上次保持以及i天卖出
dp[i][j+2] = max(dp[i-1][j+2], dp[i-1][j+1] + prices[i]);
}
}
return dp[prices.size()-1][2*k];
}
309.最佳买卖股票时机含冷冻期
看完想法:一共有4个状态,达到持有的状态,以及达到卖出/当天卖出/冷冻期状态,(因为有冷冻期,前一次的状态必须是卖出状态)。达到买入股票的状态其实有两种,一种是前一天已持有,一种是过了冷冻期/过了冷冻期保持卖出状态,这两者dp公式都需要减去prices[i]。dp初始化时,重点考虑一下dp[0][2]。最终return 时得取后面三种可能性的最大值
int maxProfit(vector<int>& prices) {
int n = prices.size();
if (n == 0) return 0;
vector<vector<int>> dp(n, vector<int>(4, 0));
dp[0][0] -= prices[0]; // 持股票
dp[0][1] = 0;//对应文中的0天处于卖出
for (int i = 1; i < n; i++) {
dp[i][0] = max(dp[i - 1][0], max(dp[i - 1][3] - prices[i], dp[i - 1][1] - prices[i]));
dp[i][1] = max(dp[i - 1][1], dp[i - 1][3]);
dp[i][2] = dp[i - 1][0] + prices[i];
dp[i][3] = dp[i - 1][2];
}
return max(dp[n - 1][3], max(dp[n - 1][1], dp[n - 1][2]));
714.买卖股票的最佳时机含手续费
看完想法:本体只需要考虑卖出时交手续费即可。相对于动态规划:122.买卖股票的最佳时机II (opens new window),本题只需要在计算卖出操作的时候减去手续费就可以了,代码几乎是一样的。
int n = prices.size();
vector<vector<int>> dp(n, vector<int>(2, 0));
dp[0][0] -= prices[0]; // 持股票
for (int i = 1; i < n; i++) {
dp[i][0] = max(dp[i - 1][0], dp[i - 1][1] - prices[i]);
dp[i][1] = max(dp[i - 1][1], dp[i - 1][0] + prices[i] - fee);
}
return max(dp[n - 1][0], dp[n - 1][1]);