动态规划算法9
LeetCode 309 买卖股票的最佳时机含冷冻期 2023.12.14
int maxProfit(vector<int>& prices) {
//1确定dp二维数组
//dp[i][0]表示遍历到第i天时持有股票的当前收入;dp[i][1]表示遍历到第i天时未持有股票的当前收入
//dp[i][2]表示遍历到第i天卖出股票的当前收入;dp[i][3]表示遍历到第i天时处于冷冻期的当前收入
vector<vector<int>> dp(prices.size(), vector<int>(4, 0));
//3初始化，第一天持有股票的收入为-prices[0]，其他值为0
dp[0][0] = -prices[0];
//2确定递推公式，4确定遍历顺序
for (int i = 1; i < prices.size(); i++)
{
//dp[i][0]表示遍历到第i天时持有股票的当前收入;
//可能值：上一天持有股票，处于不持有股票期或冷冻期购买股票
dp[i][0] = max(max(dp[i-1][0], dp[i-1][1]-prices[i]), dp[i-1][3]-prices[i]);
//dp[i][1]表示遍历到第i天时未持有股票的当前收入;
//可能值：上一天未持有股票，处于不持有股票期的冷冻期
dp[i][1] = max(dp[i-1][1], dp[i-1][3]);
//dp[i][2]表示遍历到第i天时卖出股票的当前收入;
dp[i][2] = dp[i-1][0] + prices[i];
//dp[i][3]表示第i天时处于冷冻期的当前收入，值为上一天卖出股票后的收入
dp[i][3] = dp[i-1][2];
}
//输出答案为最后一天未持有股票或卖出股票或冷冻期的收入，取最大值
return max(max(dp[prices.size()-1][1], dp[prices.size()-1][2]),dp[prices.size()-1][3]);
}
LeetCode 714 买卖股票的最佳时机含手续费 2023.12.14
int maxProfit(vector<int>& prices, int fee) {
//1确定dp二维数组
//dp[i][0]表示遍历到第i天时持有股票的当前收入;dp[i][1]表示遍历到第i天时未持有股票的当前收入
vector<vector<int>> dp(prices.size(), vector<int>(2, 0));
//3初始化，第一天持有股票的收入为-prices[0]，其他值为0
dp[0][0] = -prices[0];
//2确定递推公式，4确定遍历顺序
for (int i = 1; i < prices.size(); i++)
{
//第i天持有股票的收入=max(股票不卖(上一天持有股票的收入)，昨天没持有买完今天的股票的收入)
dp[i][0] = max(dp[i-1][0], dp[i-1][1]-prices[i]);
//第i天未持有股票的收入=max(上一天未持有股票的收入，今天卖股票(包含手续费)的收入）
dp[i][1] = max(dp[i-1][1], dp[i-1][0]-fee+prices[i]);
}
return dp[prices.size()-1][1];
}