您现在的位置是:首页 >学无止境 >LeetCode 121 买卖股票的最佳时机网站首页学无止境

LeetCode 121 买卖股票的最佳时机

yiyakaa 2024-06-17 11:27:54
简介LeetCode 121 买卖股票的最佳时机

题目: 给定一个数组 prices ,它的第 i 个元素 prices[i] 表示一支给定股票第 i 天的价格。你只能选择 某一天 买入这只股票,并选择在 未来的某一个不同的日子 卖出该股票。设计一个算法来计算你所能获取的最大利润。返回你可以从这笔交易中获取的最大利润。如果你不能获取任何利润,返回 0 。

示例 1:

输入:[7,1,5,3,6,4]
输出:5
解释:在第 2 天(股票价格 = 1)的时候买入,在第 5 天(股票价格 = 6)的时候卖出,最大利润 = 6-1 = 5 。
注意利润不能是 7-1 = 6, 因为卖出价格需要大于买入价格;同时,你不能在买入前卖出股票。

示例 2:

输入:prices = [7,6,4,3,1]
输出:0
解释:在这种情况下, 没有交易完成, 所以最大利润为 0。

思路:

dp[i][0]表示第i天持有股票的最大金额
dp[i][1]表示第i天不持有股票的最大金额
首先看dp[i][0]持有的最大金额
如果是在第i天前就持有了,那么最大金额应该是dp[i-1][0]
如果是在第i天才持有的,那么最大金额就是-prices[i]
dp[i][0] = max(dp[i - 1][0],-prices[i])
再看dp[i][1]不持有的最大金额
如果是第i天之前就不持有了,那么最大金额应该是dp[i-1][1]
如果是第i天才不持有的,那么最大金额就是dp[i-1][0]+prices[i]
dp[i][1] = max(dp[i - 1][1], dp[i - 1][0] + prices[i])
初始化:从递推公式可以看出来,和前一天的状态有关系
因此初始化前一天的就行,也就是第一天
持有:dp[0][0] = -prices[0]
不持有:dp[0][1] = 0


class Solution {
public:
	int maxProfit(vector<int>& prices) {
		if (prices.size() == 0)return 0;
		vector<vector<int>> dp(prices.size(),vector<int>(2));
		dp[0][0] = -prices[0];
		dp[0][1] = 0;
		for (int i = 1; i < prices.size();i++) {
			dp[i][0] = max(dp[i - 1][0], -prices[i]);
			dp[i][1] = max(dp[i - 1][1], dp[i - 1][0] + prices[i]);
		}
		return max(dp[prices.size()-1][0], dp[prices.size()-1][1]);
	}
};

int main() {
	vector<int> prices = { 7,1,5,3,6,4 };
	Solution ss;
	cout<<ss.maxProfit(prices)<<endl;
	return 0;
}
风语者!平时喜欢研究各种技术,目前在从事后端开发工作,热爱生活、热爱工作。