算法学习——动态规划之买卖股票问题

目录

一、买卖股票的最佳时机I（最多买卖一次）

二、买卖股票的最佳时机II（不限制买卖次数）

三、买卖股票的最佳时机三（最多买卖两次）

四、买卖股票的最佳时机四（最多买卖k次）

五、买卖股票的最佳时机含冷冻期（不限制买卖次数）

六、买卖股票的最佳时机含手续费（不限制买卖次数）

---

一、买卖股票的最佳时机I（最多买卖一次）

        原题力扣121        . - 力扣（LeetCode）

int maxProfit(vector<int>& prices)
{
	int n = prices.size();
	vector<vector<int>> dp(n, vector<int>(2));
	dp[0][0] = -prices[0];
	dp[0][1] = 0;
	for (int i = 1; i < n; i++)
	{
		dp[i][0] = max(dp[i - 1][0], -prices[i]); // 第i天持有股票时的最大现金
		dp[i][1] = max(dp[i - 1][1], dp[i - 1][0] + prices[i]); // 第i天不持有股票时的最大现金
	}
	return dp[n - 1][1];
}

二、买卖股票的最佳时机II（不限制买卖次数）

        原题力扣122        . - 力扣（LeetCode）

int maxProfitII(vector<int>& prices)
{
	int n = prices.size();
	vector<vector<int>> dp(n, vector<int>(2));
	dp[0][0] = -prices[0];
	dp[0][1] = 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]);
	}
	return dp[n - 1][1];
}

三、买卖股票的最佳时机三（最多买卖两次）

        原题力扣123        . - 力扣（LeetCode）

int maxProfitIII(vector<int>& prices)
{
	int n = prices.size();
	vector<vector<int>> dp(n, vector<int>(5));
	dp[0][0] = 0;
	dp[0][1] = -prices[0];
	dp[0][2] = 0;
	dp[0][3] = -prices[0];
	dp[0][4] = 0;
	for (int i = 1; i < n; i++)
	{
		dp[i][0] = dp[i - 1][0]; // 不操作
		dp[i][1] = max(dp[i - 1][1], dp[i - 1][0] - prices[i]); // 第i天，第一次持有股票
		dp[i][2] = max(dp[i - 1][2], dp[i - 1][1] + prices[i]); // 第i天，第一次不持有股票
		dp[i][3] = max(dp[i - 1][3], dp[i - 1][2] - prices[i]); // 第i天，第二次持有股票
		dp[i][4] = max(dp[i - 1][4], dp[i - 1][3] + prices[i]); // 第i天，第二次不持有股票
	}
	return dp[n - 1][4];
}

四、买卖股票的最佳时机四（最多买卖k次）

        原题力扣188        . - 力扣（LeetCode）

int maxProfitIV(int k, vector<int>& prices)
{
	int n = prices.size();
	vector<vector<int>> dp(n, vector<int>(2 * k + 1));
	dp[0][0] = 0;
	for (int i = 1; i < 2 * k; i += 2)
	{
		dp[0][i] = -prices[0]; // 持有时才要减去初始股
	}
	for (int i = 1; i < n; i++)
	{
		for (int j = 0; j < 2 * k; j += 2)
		{
			dp[i][j + 1] = max(dp[i - 1][j + 1], dp[i - 1][j] - prices[i]); // 第j+1次不持有
			dp[i][j + 2] = max(dp[i - 1][j + 2], dp[i - 1][j + 1] + prices[i]); // 第j+1次持有
		}
	}
	return dp[n - 1][2 * k];
}

五、买卖股票的最佳时机含冷冻期（不限制买卖次数）

        原题力扣309        . - 力扣（LeetCode）

int maxProfitFrozen(vector<int>& prices)
{
	int n = prices.size();
	vector<vector<int>> dp(n, vector<int>(4));
	dp[0][0] = -prices[0];
	dp[0][1] = 0;
	dp[0][2] = 0;
	dp[0][3] = 0;
	for (int i = 1; i < n; i++)
	{
		dp[i][0] = max(dp[i - 1][0], max(dp[i - 1][1] - prices[i], dp[i - 1][3] - 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][1], max(dp[n - 1][2], dp[n - 1][3]));
}

六、买卖股票的最佳时机含手续费（不限制买卖次数）

        原题力扣714        . - 力扣（LeetCode）

int maxProfitFee(vector<int>& prices, int fee)
{
	int n = prices.size();
	vector<vector<int>> dp(n, vector<int>(2));
	dp[0][0] = -prices[0] - fee;
	dp[0][1] = 0;
	for (int i = 1; i < n; i++)
	{
		dp[i][0] = max(dp[i - 1][0], dp[i - 1][1] - prices[i] - fee);
		dp[i][1] = max(dp[i - 1][1], dp[i - 1][0] + prices[i]);
	}
	return dp[n - 1][1];
}